この記事では、PyTorch で Tensor にディメンションを追加する方法を例に説明します。
PyTorch で特定のテンソルにディメンションを追加するにはどうすればよいですか?
ユーザーは、PyTorch の 1D テンソルや 2D テンソルなどの任意の Tensor に次元を追加できます。特定の位置でテンソルに新しい次元を追加するには、理解を深めるために次の例を確認してください。
例 1: PyTorch で 1D テンソルに次元を追加する
この例では、1D テンソルを作成し、その特定の位置に次元を追加します。実際のデモンストレーションを行うには、以下の手順に従ってください。
ステップ 1: ライブラリをインポートする
まず、トーチ ライブラリをインポートします。
輸入 松明
ステップ 2: 1D テンソルを作成する
次に、1 次元テンソルを作成します。たとえば、次のテンソルを作成し、「 バツ ' 変数：
バツ = 松明。 テンソル ( [ 5 、 3 、 8 、 2 ] )
ステップ 3: テンソル形状を表示する
次に、新しく作成したテンソル シェイプを表示して、その寸法を確認します。
以下の出力は、テンソルが 1 次元であることを示しています。
ステップ 4: 1D テンソルに次元を追加する
さあ、「」を活用してください。 torch.unsqueeze(入力、薄暗く) 」関数を使用して、1D テンソルの特定の位置に次元を追加します。たとえば、インデックス 0 でテンソルに次元を追加しています。
ここ、
- 「 新しい十 ”は追加された次元を含む変数です。
- 「 バツ ” は入力テンソルです。
- 「 ディム=0 ” はインデックス 0 に次元を追加するために使用されます。
ステップ 5: 出力を確認する
最後に、新しい次元がテンソルに追加されたかどうかを確認します。
以下の出力では、新しい次元が 0 インデックスの 1D テンソルに追加されたことがわかります。
さらに、ユーザーは他の位置に寸法を追加することもできます。ここでは、最初のインデックスにディメンションを追加しました。
例 2: PyTorch で 2D テンソルに次元を追加する
ここでは、2D テンソルを作成/作成し、特定の位置に次元を追加します。実際の実装のために、指定された手順を試してください。
ステップ 1: トーチ ライブラリをインポートする
まず、トーチ ライブラリをインポートします。
ステップ 2: 2D テンソルを作成する
次に、2 次元テンソルを作成します。たとえば、次のテンソルを作成し、「 バツ ' 変数：
ステップ 3: テンソル形状を表示する
その後、新しく作成したテンソル シェイプを表示して、その寸法を確認します。
以下の出力によると、このテンソルは 2 次元です。
ステップ 4: 2D テンソルに次元を追加する
次に、次のコマンドを使用して、2D テンソルの特定の位置に次元を追加します。 「torch.unsqueeze(input, dim)」 関数。たとえば、インデックス 0 でテンソルに次元を追加しています。
ステップ 5: 出力を確認する
最後に、新しい次元が 2D テンソルに追加されたかどうかを確認します。
以下の出力は、新しい次元がインデックス 0 の 2D テンソルに正常に追加されたことを示しています。
注記 : ここから Google Colab ノートブックにアクセスできます リンク 。
さまざまな例を使用して、PyTorch でテンソルに次元を追加する方法を効率的に説明しました。
結論
PyTorch でテンソルに次元を追加するには、まず PyTorch ライブラリをインポートします。次に、1D または 2D テンソルを作成し、その寸法を表示します。その後、「」を使用して、テンソルの特定の位置に次元を追加します。 torch.unsqueeze(入力、薄暗く) ' 関数。ユーザーは、入力テンソルと必要なインデックス位置をパラメーターとしてこの関数に渡す必要があります。この記事では、PyTorch でテンソルに次元を追加する方法を例示しました。