报错
RuntimeError: one_hot is only applicable to index tensor.
这个错误表明你正在尝试对非索引张量(index tensor)(也就是非整形张量,如浮点型)应用one_hot
函数。one_hot
函数期望输入的张量包含的是整数索引,这些索引将被转换为独热编码。
如果X
不是这样的索引或者张量,你需要先将其转换为正确的格式。例如,如果X
是一个浮点数张量,你可以使用torch.Tensor.long
或torch.Tensor.int
方法将其转换为整数张量: