张量从 (256, 256, 3) 重塑到 (1,256, 256, 3)

Question

我的问题是如何重塑张量！

这是一种语法tf.reshape(tensor, shape, name=None)

我不知道，我哪里错了，但我无法重塑

我怎样才能做到这一点？

谢谢

Answer 1

axis=0# image is your tensortf.expand_dims(your_image, axis)

Answer 2

试试下面的代码：#assuming `img` contains the data which is in the format (256,256,3)output = tf.reshape(img, [1, 256, 256, 3]) # 1 is batch size

Answer 3

您可以使用tf.expand_dims：output = tf.expand_dims(input, 0)