pytorch遇到的一些坑

pytorch遇到的一些非常害人的坑 一、tensor.scatter()暗含的坑 一、tensor.scatter()暗含的坑 tensor.scatter()方法pytorch官方文檔講解的還是很詳細的,如下圖所示: 有一個坑就是:**index參數必須是torch.int64 or torch.long!torch.int32 or torch.int都是不可以的!**否則的話,會報一些匪夷
相關文章
相關標籤/搜索