问题描述:
【功能模块】
我的网络有一个fusion_layer,用来融合(elemen-wise add)两个分支的结果。
【操作步骤&问题现象】
1、因为两个分支上特征图的分辨率不同,在融合之前需要对特征图的尺寸进行resize。我打算用ops.ResizeNearestNeighbor实现resize。
2、但是这个方法需要获取目标尺寸大小。
因为我不确定,中间过程中各分支上特征图的尺寸,所有我便不能在网络的__init__函数中预先声明这个方法了,只能在construct执行过程中需要时现声明。由于是分支,我想用CellList存放对各分支的操作。但是出现了下面的问题。
【截图信息】
下面是我做实验写的一个简单的demo。我尝试了三种形式,出现了让我费解的问题。我想知道:
CellList能否像后两种形式这样用;
有没有其他关于Resize和CellList的建议。
第一种形式:在__init__中声明,在construct中执行。正确执行,但是不能达到我的需求。