python - 如何获得张量形状python

Question

我对张量有疑问。

X.get_shape() returns (?, ?, ?, 512)

而当我尝试连接 X 和其他 vectorX=Concatenate()([X,X_add])时，它会引发异常。

层需要具有匹配形状的Concatenate输入，除了 concat 轴。得到输入形状：[(None, 32, 32, 512), (None, 256, 256, 64)]

如何获得这些形状以便在我的 resnet 中使用 Cropp2D。

标签： pythontensorflowmachine-learningkerastensor

Answer 1

有时你必须在连接之前重塑张量。但这并不能解决您的问题，因为这两个张量的大小不同：32*32*512 != 256* 256*64。