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

Question

我的问题是如何重塑张量！这是一种语法tf.reshape(tensor, shape, name=None)我不知道，我哪里错了，但我无法重塑我怎样才能做到这一点？谢谢

Answer 1

axis=0

# image is your tensor

tf.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)