勉強メモ】pytorchのsqueeze()関数とunsqueeze()関数の紹介

I. unsqueeze()関数

1. まず最初に

aの次元が(2, 3)であることがわかる。

2. 2次元目に次元を追加して、その次元が（2, 1, 3）になるようにする。

a の次元が (2, 1, 3) に変更されたことがわかりますが、この場合も、1 つ下の次元に次元を追加する必要がある場合は、b.unsqueeze(-2) を使用します。

2つ目は、squeeze()関数の紹介です。

1. まず、次元(1, 2, 3)のテンソル(tensor)を取得する。

cの寸法は、図のように（1、2、3）です。

2. 以下は、squeeze()関数を用いて1次元目を削除しています。

ご覧の通り、ディメンションは(2, 3)に変更されました。

3. さらに

次元が変わっていないことがわかります。（1、2、3）のままです。これは、次元が1のときだけ削除されるからです。