关于flatten
收藏
现在有数据维度
[4096,1,5,5]
在做完卷积后要接入全连接层
需要flatten,但是如何实现第3-4维度的flatten?类似 pytorch.flatten(stardim=1) 的实现效果
看文档没太明白,没找到比较全的资料,全解
0
收藏
请登录后评论
self.conv2d = Conv2D(num_channels=1, num_filters=1, filter_size=3, stride=1)
self.pool2d = Pool2D(pool_size=2, pool_type='max', pool_stride=1, global_pooling=False)
yourenma
用fluid.layers.reshape()