PyTorchでテンソルのサイズを変更するにはどうすればよいですか？

PyTorchテンソルのサイズを変更するには、 .view（）を使用します 方法。テンソルの次元を増減することはできますが、テンソル内の要素の総数がサイズ変更の前後で一致する必要があることを確認する必要があります。

ステップ

• 必要なライブラリをインポートします。以下のすべてのPythonの例では、必要なPythonライブラリはトーチです。 。すでにインストールされていることを確認してください。

• PyTorchテンソルを作成して印刷します。

• .view（）を使用して、上記で作成したテンソルのサイズを変更します 値を変数に割り当てます。 .view（） 元のテンソルのサイズを変更しません。名前が示すように、新しいサイズのビューのみが表示されます。

• 最後に、サイズ変更後にテンソルを印刷します。

例1

＃PyTorchでテンソルのサイズを変更するPythonプログラム＃ライブラリのインポートimport torch＃テンソルの作成T =torch.Tensor（[1、2、3、4、5、6]）print（T）＃Tのサイズを2x3xに変更=T.view（2,3）print（ "Tensor after resize：\ n"、x）＃Tを2x3xにサイズ変更する別の方法=T.view（-1,3）print（ "Tensor after resize：\ n"、 x）＃他の方法でTを2x3xにサイズ変更=T.view（2、-1）print（ "Tensor after resize：\ n"、x）

出力

上記のPython3コードを実行すると、次の出力が生成されます

 tensor（[1.、2.、3.、4.、5.、6.]）resize：tensor（[[1.、2.、3.]、[4.、5.、 6。]]）resize：tensor（[[1.、2.、3。]、[4.、5.、6.]]）後のテンソルresize：tensor（[[1.、2.、3 。]、[4.、5.、6。]]）

例2

＃ライブラリをインポートするimport torch＃テンソル形状を作成する4x3T =torch.Tensor（[[1,2,3]、[2,1,3]、[2,3,5]、[5,6,4 ]]）print（T）＃Tのサイズを3x4x =T.view（-1,4）print（ "Tensor after resize：\ n"、x）＃Tを3x4x =T.view（3、 -1）print（ "サイズ変更後のテンソル：\ n"、x）＃Tのサイズを2x6xに変更=T.view（2、-1）print（ "サイズ変更後のテンソル：\ n"、x）

出力

上記のPython3コードを実行すると、次の出力が生成されます

 tensor（[[1.、2.、3.]、[2.、1.、3.]、[2.、3.、5.]、[5.、6.、4.]] ）resize：tensor（[[1.、2.、3.、2。]、[1.、3.、2.、3。]、[5.、5.、6.、4.]]後のテンソル）resize：tensor（[[1.、2.、3.、2。]、[1.、3.、2.、3。]、[5.、5.、6.、4.]]後のテンソル）resize：tensor（[[1.、2.、3.、2.、1.、3.]、[2.、3.、5.、5.、6.、4.]]）<後のテンソル/ pre>