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

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

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

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

 

 

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