当分配给变量时,张量的输出形状会改变

2024-06-10 21:14:13 发布

您现在位置:Python中文网/ 问答频道 /正文

定义了一个模型,输入形状(96,96,3)并输出形状(128),即对输入大小的给定图像给出128大小的嵌入。类似于facenet

createmodel()是一个函数,它在构造完整模型后返回完整模型。由于模型期望以(批量大小,96,96,3)的形式输入,因此我传递3个图像,即(3,96,96,3),因此模型的输出将是形状(1,3128)。记住这一点,请考虑以下几点:

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)

当指定给变量时,为什么输出张量的形状会改变?你知道吗


Tags: 函数模型图像定义outputsprints形状tensor
1条回答
网友
1楼 · 发布于 2024-06-10 21:14:13

Relevant post

你知道吗tf.形状(input,name=None)返回表示输入形状的一维整数张量。你知道吗

使用x.get_shape()返回x变量的张量形状。你知道吗

相关问题 更多 >