import torch
num_class = 5
N = 3
tensor = torch.randint(0, num_class, [N])
print(tensor)
one_hot = torch.zeros(N, num_class).long()
one_hot.scatter_(dim=1,index=tensor.unsqueeze(dim=1),src=torch.ones(N, num_class).long())
print(one_hot)

本文介绍如何使用PyTorch实现One-Hot编码,包括初始化张量、使用scatter_函数进行转换的过程。One-Hot编码是机器学习中常用的预处理手段,将类别变量转换为二进制向量。
import torch
num_class = 5
N = 3
tensor = torch.randint(0, num_class, [N])
print(tensor)
one_hot = torch.zeros(N, num_class).long()
one_hot.scatter_(dim=1,index=tensor.unsqueeze(dim=1),src=torch.ones(N, num_class).long())
print(one_hot)


被折叠的 条评论
为什么被折叠?
