在keras的网络中,如果用layer_name.shape的方式获取shape信息将会返还tensorflow.python.framework.tensor_shape.TensorShape
其中包含的是tensorflow.python.framework.tensor_shape.Dimension
正确的方式是使用
import keras.backend as K
K.int_shape(laye_name)
————————————————
版权声明:本文为CSDN博主「weixin_39165863」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/weixin_39165863/article/details/90715965
本文介绍在Keras框架中正确获取网络层形状的方法。使用`keras.backend.int_shape()`替代直接通过`.shape`访问,以避免返回TensorFlow的`TensorShape`对象,确保得到易于操作的形状数据。

603

被折叠的 条评论
为什么被折叠?



