pytorch 独热编码报错的解决办法:one_hot is only applicable to index tensor


首先,报错原因,我认为是数据类型错误,

在文档中表示,第一个tensor参数的数据类型为LongTensor,也就是torch.int64类型的,如果你有报这个错:“one_hot is only applicable to index tensor”,可以查看一下你传入的参数是不是int32或者其他类型的,如果是的话,强制类型转换更改一下就好了,也就是说改成int64的。

例如下面的代码:第一行进行了强制类型转换,后面就不报错了

 

 

如果这篇文章对你有帮助,希望能收到你的一个点赞!


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM