首先，報錯原因，我認為是數據類型錯誤，
在文檔中表示，第一個tensor參數的數據類型為LongTensor，也就是torch.int64類型的，如果你有報這個錯：“one_hot is only applicable to index tensor”，可以查看一下你傳入的參數是不是int32或者其他類型的，如果是的話，強制類型轉換更改一下就好了，也就是說改成int64的。
例如下面的代碼：第一行進行了強制類型轉換，后面就不報錯了
如果這篇文章對你有幫助，希望能收到你的一個點贊！
免責聲明！
本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。