import torch import numpy as np
x = np.array([1, 2, 3]) x_tensor = torch.from_numpy(x) y = x_tensor.reshape(-1, 1) y_one_hot = torch.zeros(3, 5) y_one_hot.scatter_(1, y, 1) print(y_one_hot)
运行结果在这里: 大佬们,为什么会出现这种问题啊,有没有解决方法!!!