下面的程序會報錯IndexError: tensors used as indices must be long, byte or bool tensors

mask = torch.Tensor([True,True,False]) a = torch.Tensor([3,2,1]) a[mask]=0 print(a)

原因是索引要為long, byte 或 bool類型，因此需要將mask轉換為bool類型，如下：

mask = torch.Tensor([True,True,False]).type(torch.bool) a = torch.Tensor([3,2,1]) a[mask]=0 print(a) tensor([0., 0., 1.])

?

總結

以上是生活随笔為你收集整理的IndexError: tensors used as indices must be long, byte or bool tensors的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。