- 第一个参数是代表横轴删除还是纵轴删除,0为横轴,1为纵轴
- 第二个和第三个参数代保留开始轴到结束轴的数字,类似于切片
x = torch.tensor([[1,2,3],[4,5,6],[7,8,9]])
x
tensor([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
x.narrow(0,0,2)
tensor([[1, 2, 3],
[4, 5, 6]])
x.narrow(1,1,2)
tensor([[2, 3],
[5, 6],
[8, 9]])