压缩一层的输出

时间:2019-01-25 21:40:49

标签: tensorflow

我试图从具有单例尺寸的一层压缩输出,但是它不起作用。

这是我的代码:

p3 = MaxPooling3D((2, 2, 2))(c3)

print(p3.shape.dims)


tf.squeeze(p3)

print(p3.shape.dims)

挤压操作前后的形状相同

  

[Dimension(None),Dimension(1),Dimension(64),Dimension(64),Dimension(32)]

1 个答案:

答案 0 :(得分:0)

我无法验证它,因为您没有发布代码,但是您可能需要将挤压结果分配给某些东西,然后打印出它的形状。

如果这样做没有帮助,请更新您的代码段以包含更多详细信息,例如您是否以急切模式运行或其他事情