torch.narrow_copy¶
- torch.narrow_copy(input, dim, start, length, *, out=None) Tensor ¶
与
Tensor.narrow()
相同,除了它返回一个副本而不是共享存储。这主要用于稀疏张量,它们没有共享存储的窄方法。- 参数
- 关键字参数
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()
用于非副本变体