摘要: 首先,报错原因,我认为是数据类型错误, 在文档中表示,第一个tensor参数的数据类型为LongTensor,也就是torch.int64类型的,如果你有报这个错:“one_hot is only applicable to index tensor”,可以查看一下你传入的参数是不是int32或者其 阅读全文
posted @ 2020-12-03 16:35 阳光中的影子 阅读(16560) 评论(0) 推荐(2) 编辑