pytorch遇到的一些坑

pytorch遇到的一些非常害人的坑 一、tensor.scatter()暗含的坑 一、tensor.scatter()暗含的坑 tensor.scatter()方法pytorch官方文档讲解的还是很详细的,如下图所示: 有一个坑就是:**index参数必须是torch.int64 or torch.long!torch.int32 or torch.int都是不可以的!**否则的话,会报一些匪夷
相关文章
相关标签/搜索