2019-06-28こしあんTweetPocket出力が複数あるモデルの訓練というのは少し複雑なモデルだとよく出てきます。PyTorchでは複数出力のモデルの、出力の型はどうなっているでしょうか。中間層の値を取りたい場合も使えます。conv3という複数の畳み込み層があります。各中間層の出力をまとめて返すモデルを試しに作ってみます。結論からいうと、forwardの返り値の定義に依存します。上のコードのように、forwardのreturnをlistとして定義すると、出力はtorch.Tensorのlistになります。Tupleにしてみる出力もTupleになります。forで取り出せるのは変わりませんね。'torch.Tensor'>つまりは、モデル側のforwardで定義した通りに帰ってくるということです。Commentコメントをキャンセルメールアドレスが公開されることはありません。