翻译:Torch DoubleTensor和CharTensor不兼容。

如何将CharTensor掩码与DoubleTensor相乘?我正在为torch制作一个模块,它应该类似于ReLU模块。因此,self.mask应该是一个由0和1组成的掩码,我想将其应用于DoubleTensor self.output。目前self.mask是一个CharTensor,我无法像日志中所示那样将两者相乘。这似乎是如此微不足道,但我已经在网上找了很长时间,无法找到解决方案。

无限的感谢,

-一个焦虑的编码者

点赞
用户1417179
用户1417179

Torch 提供了不需要手动计算乘法的方法。其中最简单的方法可能是反转掩码,然后使用 索引 [] 操作符

self.mask = torch.le(input, 0)
self.output[self.mask] = 0
2015-10-16 16:36:57