torch.narrow()基本用法介绍

torch.narrow(input,dim,start,length)

• 从 input 张量中返回一个范围限制后的 张量，范围限制条件为：沿维度dim 从 start 到start+length 的范围区间，类似于数组切片用法，返回的张量与 input 张量共享相同储存基础

参数

• input(Tensor) ，需处理的张量；
• dim(int)，沿着限制的轴；
• start(int) ，张量起始点；
• length(int) ，缩窄长度;

例子如下：

rand_float = torch.randn((5,3))# 随机生成 5*3数据
rand_float
>>>
tensor([[-0.4972, -0.1363, -1.8918],
        [ 1.2994, -1.0091,  0.1862],
        [ 0.5525,  1.3073,  1.3741],
        [-1.7242, -0.3593, -0.7546],
        [-0.3328,  0.3333,  0.0096]])
        
rand_float.narrow(0,1,2)# 沿第一维度开始，第一行为开始，长度为2
>>>
tensor([[ 1.2994, -1.0091,  0.1862],
        [ 0.5525,  1.3073,  1.3741]])