
一_hot编码:`torch.Tensor.scatter_()`函数详解
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本文详细介绍Python深度学习库PyTorch中的`torch.Tensor.scatter_()`函数,探讨其在热编码和数据操作中的应用。
`torch.Tensor.scatter_()` 是 `torch.gather()` 函数的逆向操作。这两个函数可以看作是一对兄弟函数:gather 用于解码 one-hot 编码,而 scatter_ 则用于生成 one-hot 编码。具体来说,scatter_(dim, index, src) 的参数包括:
- dim (python:int): 进行寻址的坐标轴。
- index (LongTensor): 索引位置。
- src(Tensor): 用来散布(scatter)值的源张量;如果未指定 value,则使用此源张量中的元素作为需要散布的值。
下面通过一个例子来具体说明如何使用。
全部评论 (0)
还没有任何评论哟~


