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