基本上如何将CharTensor蒙版与DoubleTensor相乘?我为火炬制作模块,它应该与ReLU模块类似。因此,self.mask应该是0和1的掩码,我想用DoubleTensor self.output。目前self.mask是一个CharTensor,我无法将两者相乘,如日志中所示。这似乎是微不足道的,但我现在一直在线寻找,无法找到解决方案。
无限的谢谢,
- 焦虑的编码员
函数ReQU:updateOutput(输入)
...
self.mask = torch.gt(输入,0)
self.output:CMUL(self.mask)
...
日志: 无效参数:DoubleTensor ByteTensor 预期参数: DoubleTensor [DoubleTensor] DoubleTensor 堆栈追溯: [C]:在函数'cmul'中 ./requ.lua:21:在函数'forward'
答案 0 :(得分:2)
Torch公开了这样做的方法,而无需自己乘以值。最简单的可能是反转你的面具并使用index []
operator:
self.mask = torch.le(input, 0)
self.output[self.mask] = 0