one hot编码：`torch.Tensor.scatter_()`函数用法详解

miguemath

于 2020-02-17 12:55:21 发布

阅读量3.8k

点赞数 5

分类专栏： Pytorch Python 文章标签： python pytorch 机器学习深度学习

本文链接：https://blog.csdn.net/wangkaidehao/article/details/104356425

版权

本文详细介绍了如何使用PyTorch中的torch.Tensor.scatter_()函数进行One Hot编码，分别阐述了dim=0和dim=1的情形，并通过实例解析了该函数的工作原理和使用方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

torch.Tensor.scatter_()是torch.gather()函数的方向反向操作。两个函数可以看成一对兄弟函数。gather用来解码one hot，scatter_用来编码one hot。

scatter_(dim, index, src) → Tensor

dim (python:int) – 用来寻址的坐标轴
index (LongTensor) – 索引
src(Tensor) –用来scatter的源张量，以防value未被指定。
value(python:float) – 用来scatter的源张量，以防src未被指定。

现在我们来看看具体这么用，看下面这个例子就一目了然了。

dim =0

import torch

x = torch.tensor([[0.9413, 0.9476, 0.1104, 0.9898, 0.6443],
            [0.6913, 0.8924, 0.7530, 0.8874, 0.0557]])

result = torch.zeros(3, 5)
indices = torch.tensor([[0

最低0.47元/天解锁文章