tensor.narrow()函数

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.]])