RuntimeError: Expected object of type Variable[torch.cuda.FloatTensor] but found type Variable[torch.cuda.LongTensor] for argument #1 'other'
两个变量类型不一致,将longtensor改为floattensor即可解决.
mask = mask.float()
附上参考博客,写得很详细.
RuntimeError: Expected object of type Variable[torch.cuda.FloatTensor] but found type Variable[torch.cuda.LongTensor] for argument #1 'other'
两个变量类型不一致,将longtensor改为floattensor即可解决.
mask = mask.float()
附上参考博客,写得很详细.