def meshgrid(height, width):
x = tf.tile(tf.linspace(-1.,1.,width), [height])
y = repeat(tf.linspace(-1.,1.,height), width)
return x, y
微信
公众号
扫码关注公众号