Изменение тензора с (256, 256, 3) на (1,256, 256, 3)

Question

Мой вопрос: как мне изменить форму тензора!

Вот один синтаксис tf.reshape(tensor, shape, name=None)

Я не знаю, где я ошибаюсь, но я не могу изменить форму

как я могу это сделать?

спасибо

Answer 1

Попробуйте следующий код:

#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 2

axis=0

# image is your tensor

tf.expand_dims(your_image, axis)

Answer 3

Вы можете использовать tf.expand_dims:

output = tf.expand_dims(input, 0)