タグ: 小ネタ

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

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

条件に応じた配列の要素の抽出をTensorFlowで行う

10k{icon} {views} Numpyで条件を与えて、インデックスのスライスによって配列の要素を抽出する、というようなケースはよくあります。これをTensorFlowのテンソルでやるのにはどうすればいいのでしょう […]...

TensorFlowで値のソートをする方法

4.3k{icon} {views} TensorFlowでNumpyのnp.sortやnp.argsortのようなソートを行うことを考えます。一般にTensorFlowで値のソートというと、自動微分もあわさって難しいよ […]...