使用动态图定义网络时，先在构造函数里定义各个子网络，如何在网络的构造函数里获得特征图尺寸？
因为一些网络结构在压缩或膨胀张量时会计算特征图的尺寸
特征图尺寸在forward里很好的到，怎么在构造函数里得到呢？
动态图里能得到么？
我想问的是怎么画中间的特征图？
在forward里打印不行么？
可以用visualdl显示
好像看过ai studio上的例子
但是是要在代码里加入的
我是想用在模型结构里。
non-local模块里需要传入特征图的尺寸做参数，所以在定义模型结构类的构造函数里就得得到特征图的尺寸。
我现在只能是，自己算出来在non-local模块里写死。
我想写成一个通用模块
summary函数可以输出每层的shape
用low api就好了
summary是可以查看。
我是想将上一层特征图的形状当参数传给下一层
我觉得我用的api已经不是high的了吧，难道还有比Layer更low的api？
因为一些网络结构在压缩或膨胀张量时会计算特征图的尺寸
特征图尺寸在forward里很好的到，怎么在构造函数里得到呢？
动态图里能得到么？
我想问的是怎么画中间的特征图？
在forward里打印不行么？
可以用visualdl显示
好像看过ai studio上的例子
但是是要在代码里加入的
我是想用在模型结构里。
non-local模块里需要传入特征图的尺寸做参数，所以在定义模型结构类的构造函数里就得得到特征图的尺寸。
我现在只能是，自己算出来在non-local模块里写死。
我想写成一个通用模块
summary函数可以输出每层的shape
用low api就好了
summary是可以查看。
我是想将上一层特征图的形状当参数传给下一层
我觉得我用的api已经不是high的了吧，难道还有比Layer更low的api？