форма вывода тензорных изменений при назначении переменной

Question

Определяет модель, которая принимает ввод формы (96,96,3) и выводит форму (128), т. Е. Дает вложение размером 128 для данного изображения входного размера. Похож на facenet

createmodel () - это функция, которая возвращает полную модель после ее построения. Так как модель ожидает ввода в виде (batch_size, 96,96,3), я пропускаю 3 изображения, т.е. (3,96,96,3), поэтому выходные данные модели будут иметь форму (1,3,128). Учитывая это, учтите следующее:

nn4_small2=createmodel()
print(nn4_small2.outputs[0][2])

#prints Tensor("strided_slice_11:0", shape=(128,), dtype=float32)

x=nn4_small2.outputs[0][2]
print(tf.shape(x))

#prints Tensor("Shape_6:0", shape=(1,), dtype=int32)

Почему форма выходного тензора изменяется при назначении переменной?

Answer 1

Соответствующий пост

tf.shape (input, name = None) возвращает 1-D целочисленный тензор, представляющий форму ввода.

использование x.get_shape () возвращает TensorShape переменной x.