input.narrow(dimension, start, length) → Tensor
# 表示取变量input在第dimension维上,从索引start到start+length范围(不包括start+length)的值。
example:
In [2]: x = torch.Tensor([[1,2,3], [4,5,6], [7,8,9]])
In [4]: x.narrow(0,0,3)
Out[4]:
tensor([[1., 2., 3.],
[4., 5., 6.],
[7., 8., 9.]])
In [6]: x.narrow(0,1,2)
Out[6]:
tensor([[4., 5., 6.],
[7., 8., 9.]])
In [7]: x.narrow(1, 1, 2)
Out[7]:
tensor([[2., 3.],
[5., 6.],
[8., 9.]])
tensor.narrow()函数
最新推荐文章于 2023-12-07 18:04:59 发布