【ひとり読書会】PyTorch&深層学習プログラミング（第11回）

どうも、カタミチです。

さて、今日も「最短コースでわかる PyTorch&深層学習プログラミング」のひとり読書会、やっていきますかねー。

今日は2-2「テンソル」の続きですね。

2-2. テンソル（続き）

テンソルの各要素の型は浮動小数点数型である「float32」にしておく…ってのがここまでのテンソルの話でしたが、整数値型で格納しておいた方が都合がいいケースもあるようです。そのために使うのは「long関数」ですね。

私の知っている言葉でいうと、型変換（キャスト）ってやつですかね。long関数を噛ませた整数型は「int64」と表記されるようですね。

…まぁ、上の変換はあくまでも、これまで使った変数（r1）を流用したくてこんな書き方になっていますが、整数型で定義したいのなら初めからlong関数で定義しておけば良さそうですね。

次に「view関数」についてです。前回例で使った3階テンソルである変数r3を流用します。

view関数は、テンソルの階数を変換するときに使う関数のようです。上の（3, 2, 2）の3階テンソルを、（3, 4）の2階テンソルに変換する場合…

こう書きます。

重要なルールは「変換元の要素の数と変換先の要素の数が一致する」ということです。元の要素数は、\(3 \times 2 \times 2 = 12\)ってことになりますので、2階テンソルの行の数が\(3\)であるならば、列の数は\(4\)ってことになります。\(3 \times 4 = 12\)ですからね。

さてコードの書き方ですが、本当は「r6 = r3.view(3, 4)」って書く方が厳密なんですが、「\(3\)にいくつをかければ\(12\)になるっけ？」というのを導出するのが苦手な人向けに、「-1って書いとけば勝手に数値を導出しておくよー」という記法があるようです。…まぁ、苦手じゃなくても「-1」を使っておいたほうが計算ミスが無くて無難ですかね。

ただ逆に、書いた数値が間違っていた場合にも、-1の方をいい感じにエラー無く調整してくれるので、本当に厳密に書きたいのなら使うべきではないのかもしれません。可読性も落ちますしね。要は、状況によりけり…って感じですかね。ま、そうは言いつつ-1を積極的に使うことにはなりそうです（汗）。
しかし、階数が減った場合にどういう並びになるのかは、上の結果を見てしっかり目で追っておいたほうが良さそうですね。

このview関数、いろんな階数のテンソルを1階テンソルにする時に良く使うみたいです。