如何在tensorflow中创建特定形状的1 ?

Question

我需要创建一个矩形(200,240,2)例如，在numpy中我可以这样做:

r = numpy.ones_like(x) # x shape is (200,240,2)
r[...,0] = 0
r[...,1] = 1.57

如何在tensorflow中实现?

有谁能帮我一下吗?

Answer 1

你不能在tensorflow中同时做索引和赋值，但在其他情况下，你可以做与numpy相同的事情。下面是复制numpy代码的一种方法:

x_shape = x.shape
r0 = tf.zeros(x_shape[:-1])
r1 = tf.ones(x_shape[:-1]) + 0.57
r = tf.stack([r0,r1],axis=-1)

如果您需要除1或0以外的其他值，您可以进行简单的算术运算:

r1 = tf.zeros(x_shape[:-1]) + 1.57