小编S. *_*ong的帖子

PyTorch 逐元素过滤层

嗨,我想添加逐元素乘法层以将输入复制到多通道,如图所示。(因此,输入大小 M x N 和乘法滤波器大小 M x N 相同),如图所示

我想添加自定义初始化值来过滤,还希望它们在训练时获得梯度。但是,我在 PyTorch 中找不到逐元素过滤层。我能做到吗?或者在 PyTorch 中是不可能的?

python neural-network python-3.x pytorch

4
推荐指数
1
解决办法
2681
查看次数

标签 统计

neural-network ×1

python ×1

python-3.x ×1

pytorch ×1