RuntimeError: one_hot is only applicable to index tensor

本文转载自查看原文 2021-11-18 14:57 3672

# 首先看传入的tensor的size，传入的tensor的维度不够则需要添加维度 # 这里的labels的size是tensor([n])的，传入需要添加维度 # 如果出现上述问题，只需要在后面加上to.(torch.int64)即可解决 # n是要分的种类
labels = torch.nn.functional.one_hot(labels.unsqueeze(0).to(torch.int64), n)

亲测有效！！！

免责声明！

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

猜您在找 『TensorFlow』one_hot化标签 Tensorflow中one_hot() 函数用法深度学习中将类别标签映射到one_hot向量 one-hot编码详解one-hot编码独热编码（One-Hot）的理解 word2vec和one-hot RuntimeError: zero-dimensional tensor (at position 0) cannot be concatenated RuntimeError: The size of tensor a (30) must match the size of tensor b (54) at non-singleton dimension 0 文本离散表示（二）：新闻语料的one-hot编码