pytorch—torch.tensor.scatter操作解析

torch.Tensor.scatter_(dim, index, src, reduce=None)

理解scatter操作:
tensor_A.scatter_(dim, index, tensor_B): tensor_B的每个元素，都按照 index 被scatter（可以理解为填充）到目标tensor_A中。

(1) index和源tensor_B维度一致;

(2) tensor_A一般是全零的张量，其某些特定位置的值由 tensor_B 中的值填充。

(3) 注意如何根据index选取tensor_B中的值：

对于2-D tensor:

if dim=0, tensor_A[index[i][j]][j] = tensor_B[i][j];if dim=1, tensor_A[i][index[i][j]] = tensor_B[i][j];

对于3-D tensor：

if dim = 0，tensor_A[index[i][j][k]][j][k] = tensor_B[i][j][k]
if dim = 1，tensor_A[i][index[i][j][k]][k] = tensor_B[i][j][k]
if dim = 2，tensor_A[i][j][index[i][j][k]] = tensor_B[i][j][k]

举例：

如果对您有帮助，麻烦点赞关注，这真的对我很重要！！！如果需要互关，请评论或者私信！

pytorch—torch.tensor.scatter操作解析相关推荐

Pytorch——torch.Tensor.data操作的含义
1 致谢感谢陈助教的帮助! 2 前言今天在学习FaceBoxes~ 看到一句代码不是很懂, prior_data = priors.data 因为data是torch.Tensor类的一个成员变量 ...
pytorch torch.Tensor.new_ones()（返回一个与size大小相同的用1填充的张量。默认返回的Tensor具有与此张量相同的torch.dtype和torch.device）
from https://pytorch.org/docs/1.1.0/tensors.html?highlight=new_ones#torch.Tensor.new_ones new_ones(s ...
pytorch torch.Tensor.numpy()（从张量创建一个numpy数组，数组和张量共享相同内存）
https://pytorch.org/docs/1.1.0/tensors.html?highlight=numpy#torch.Tensor.numpy numpy() → numpy.ndarr ...
pytorch torch.Tensor.clone()（返回张量自身的副本。副本具有与自身相同的大小和数据类型。）
https://pytorch.org/docs/1.1.0/tensors.html?highlight=clone#torch.Tensor.clone clone() → Tensor Retu ...
Pytorch(2)-tensor常用操作
tensor常用数学操作 1. 随机数 1.1 torch.rand() - 均匀分布数字 1.2 torch.randn() - 正态分布数字 2. 求和 2.1 torch.sum(data, d ...
PyTorch | torch.tensor使用方法，如何使用torch.tensor
公众号[计算机视觉联盟]后台回复[PyTorch]可以获得PyTorch学习教程pdf版通过torch.tensor创建张量 arr = np.ones((3,3)) print("由np ...
pytorch torch.Tensor.view
应用 >>> x = torch.randn(4, 4) >>> x.size() torch.Size([4, 4]) >>> y = x.vi ...
pytorch torch.tensor
应用 torch.tensor([[0.1, 1.2], [2.2, 3.1], [4.9, 5.2]]) ---------------------------------------------- ...
PyTorch——torch.Tensor与np.ndarray（NumPy）之间的类型转换
1 前言今天在写 Digit Recognizer的代码~ 在对提交文件submission.csv进行写入操作的时候,总会有报错,其中一个原因是其实pandas对np.ndarray数据更加友好, ...
PyTorch torch.Tensor.contiguous() 用法与理解
中文文档: contiguous() → Tensor 返回一个内存连续的有相同数据的 tensor,如果原 tensor 内存连续则返回原 tensor 英文文档: contiguo ...

pytorch—torch.tensor.scatter操作解析

pytorch—torch.tensor.scatter操作解析相关推荐

最新文章

热门文章