如何在张量流中创建特定形状的形状？

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