快捷方式

torch.Tensor.index_copy_

Tensor.index_copy_(dim, index, tensor) Tensor

通过按 index 中给定的顺序选择索引,将 tensor 的元素复制到 self 张量中。例如,如果 dim == 0index[i] == j,则 tensor 的第 i 行将复制到 self 的第 j 行。

tensor 的第 dim 维的大小必须与 index 的长度(必须是向量)相同,并且所有其他维度都必须与 self 匹配,否则将引发错误。

注意

如果 index 包含重复的条目,则来自 tensor 的多个元素将被复制到 self 的相同索引。结果是不确定的,因为它取决于最后一次复制发生的时间。

参数
  • dim (int) – 要索引的维度

  • index (LongTensor) – 要从中选择的 tensor 的索引

  • tensor (Tensor) – 包含要复制的值的张量

示例

>>> x = torch.zeros(5, 3)
>>> t = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]], dtype=torch.float)
>>> index = torch.tensor([0, 4, 2])
>>> x.index_copy_(0, index, t)
tensor([[ 1.,  2.,  3.],
        [ 0.,  0.,  0.],
        [ 7.,  8.,  9.],
        [ 0.,  0.,  0.],
        [ 4.,  5.,  6.]])

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源