pytorch遇到的一些坑

一、tensor.scatter()暗含的坑

tensor.scatter()方法pytorch官方文档讲解的还是很详细的，如下图所示：

有一个坑就是：**index参数必须是torch.int64 or torch.long！torch.int32 or torch.int都是不可以的！**否则的话，会报一些匪夷所思的错误！如下图所示：