как конвертировать (1084, 1625, 3) в (нет, нет, нет, 3)

Question

Здесь у меня есть один тензор с формой (1084, 1625, 3).

И мне нужно изменить его на (нет, нет, нет, 3).

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

Я использовал этот код, но он не работает.

image = tf.cast(img, tf.float32)

image = (image / 127.5) - 1

Answer 1

Я не думаю, что вы можете сделать это. Я думаю, что вы пытаетесь превратить 3D-тензор в 4D-тензор. Я предполагаю, что это источник вашей проблемы. Вы можете сделать это, чтобы добавить 4-е измерение, потому что это нужно Tensorflow:

import tensorflow as tf

tensor = tf.random.uniform((100, 100, 3), 0, 256, dtype=tf.int32)

new = tf.expand_dims(tensor, axis=0)

print(new.shape)

Out[14]: TensorShape([1, 100, 100, 3])

Но тогда я могу ошибаться. Если это так, вы можете предоставить свою трассировку ошибки и код.