print(torch.nn.functional.one_hot(t, num_classes=7))
有个坑,使用的时候必须转换为 torch.int64 类型,不然会报错
t = t.to(torch.int64)
wg_idBccacB ©著作权
文章标签 pytorch函数用法 文章分类 PyTorch 人工智能
print(torch.nn.functional.one_hot(t, num_classes=7))
有个坑,使用的时候必须转换为 torch.int64 类型,不然会报错
t = t.to(torch.int64)
上一篇:git clone 加速
pytorch,CUDA是否可用,查看显卡显存剩余容量
One-Hot 编码1. F.one_hotpytorch 现在自带的将标签转成one-hot编码方法import torch.nn.functional a
举报文章
请选择举报类型
补充说明
0/200
上传截图
格式支持JPEG/PNG/JPG,图片不超过1.9M