カテゴリー: Python

tf.tensordotで行列積を表現するための設定

4.2k{icon} {views} TensorFlowのtensordotという関数はとても強力で、テンソルに対する行列積に対する計算をだいたい表現できます。しかし、軸の設定がいまいちよくわからなかったので、確かめて […]...

PyTorchで複数出力があるモデルの出力の型について

8k{icon} {views} 出力が複数あるモデルの訓練というのは少し複雑なモデルだとよく出てきます。PyTorchでは複数出力のモデルの、出力の型はどうなっているでしょうか。それを見ていきます。中間層の値を取りたい […]...

PyTorchで行列(テンソル)積としてConv2dを使う

4.2k{icon} {views} PyTorchではmatmulの挙動が特殊なので、思った通りにテンソル積が取れないことがあります。この記事では、基本的な畳み込み演算である「Conv2D」を使い、Numpyのドット積 […]...