2024年1月15日
将array转化为tensor，再转化为独热编码
摘要： # 将向量转化为tensor编号，并转化为独热编码 m = np.array([0, 1,2,3,4]) x = torch.Tensor(m) print(x) b = F.one_hot(x.unsqueeze(0).to(torch.int64), 5) print(b) 参考链接 https 阅读全文
posted @ 2024-01-15 16:49 芋圆院长 阅读(34) 评论(0) 推荐(0) 编辑