1. Learn
    2. /
  2. 课程
    3. /
  3. 使用 PyTorch 的深度学习入门
Connected

道练习

创建 one-hot 编码的标签

One-hot 编码会将一个整数标签转换为一个包含 N 个元素的向量，其中 N 为类别数。该向量在正确的位置是 1，其余位置为 0。

在本练习中，您将先为 y 手动创建一个 one-hot 向量，然后使用 PyTorch 来简化这一过程。您的数据集共有 3 个类别（0、1、2）。

numpy（np）、torch.nn.functional（F）和 torch 已为您导入。

说明

100 XP
  • 使用提供的 NumPy 数组手动对真实标签 y 进行 one-hot 编码，并保存为 one_hot_numpy。
  • 使用 PyTorch 对 y 进行 one-hot 编码，并保存为 one_hot_pytorch。