Introdução ao uso básico de torch.narrow ()

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

• A partir do inputtensor retorna um tensor para limitar o âmbito dos limites do intervalo é: ao longo de uma dimensão dima partir startda start+lengthgama do intervalo, semelhante ao uso de faixa da matriz, o tensor devolvido inputpara partilhar o mesmo tensor de base de armazenamento

parâmetro

• input(Tensor) , O tensor a ser processado;
• dim(int), Ao longo do eixo de restrição;
• start(int) , O ponto de partida do tensor;
• length(int) , Estreite o comprimento;

Os exemplos são os seguintes:

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