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