トーチのチュートリアルで、次の行を見つけました。
mean[i] = trainData.data[{ {},i,{},{} }]:mean()
インデックス作成 { {},i,{},{} } が何をしているのか説明できる人はいますか? 推測はできましたが、正確なメカニズムを知りたかったのです。
前もって感謝します。
これは実際にはテンソルのナローイング/スライシングの簡潔な構文であり、ドキュメントで詳しく説明されています。
内[{ ... }]
では、テンソルの各次元に対して次のことができます。
n
のみを保持する数値を渡します。n
{start,end}
すべてのコンポーネントをこの次元に沿っstart
て保持する範囲を渡します。end
{}
この次元に沿ってすべてのコンポーネントを保持するために渡します。この正確なケースでは、2 番目の次元に沿って 番目のコンポーネントのみを保持することにより、u * v * w * x
テンソルからテンソルに 狭められます。u * 1 * w * x
i