快捷方式

torch.narrow_copy

torch.narrow_copy(input, dim, start, length, *, out=None) Tensor

Tensor.narrow() 相同,除了它返回一个副本而不是共享存储。这主要用于稀疏张量,它们没有共享存储的窄方法。

参数
  • input (Tensor) – 要缩小的张量

  • dim (int) – 要缩小的维度

  • start (int) – 要开始缩小维度的元素索引。可以为负数,表示从 dim 的末尾索引

  • length (int) – 缩小维度的长度,必须为弱正数

关键字参数

out (Tensor, optional) – 输出张量。

示例

>>> x = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
>>> torch.narrow_copy(x, 0, 0, 2)
tensor([[ 1,  2,  3],
        [ 4,  5,  6]])
>>> torch.narrow_copy(x, 1, 1, 2)
tensor([[ 2,  3],
        [ 5,  6],
        [ 8,  9]])
>>> s = torch.arange(16).reshape(2, 2, 2, 2).to_sparse(2)
>>> torch.narrow_copy(s, 0, 0, 1)
tensor(indices=tensor([[0, 0],
                       [0, 1]]),
       values=tensor([[[0, 1],
                       [2, 3]],

                      [[4, 5],
                       [6, 7]]]),
       size=(1, 2, 2, 2), nnz=2, layout=torch.sparse_coo)

另请参阅

torch.narrow() 用于非副本变体

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源