Pytorch报错RuntimeError: one_hot is only applicable to index tensor.

报错

RuntimeError: one_hot is only applicable to index tensor.

这个错误表明你正在尝试对非索引张量(index tensor)(也就是非整形张量,如浮点型)应用one_hot函数。one_hot函数期望输入的张量包含的是整数索引,这些索引将被转换为独热编码。

如果X不是这样的索引或者张量,你需要先将其转换为正确的格式。例如,如果X是一个浮点数张量,你可以使用torch.Tensor.longtorch.Tensor.int方法将其转换为整数张量:

你可能感兴趣的:(Solve,Problems,pytorch,人工智能,python)