トーチでバッチ処理に使用する入力の次元はどれですか?トーチモジュールに供給する前に入力をバッチする必要がありますか?
私はトレーニングのための1000の例があり、各トレーニングの例は10 * 5の次元です。さて、私は100の例のバッチとしてこのデータをSequencerに送りたいと思っています。
入力はどのように構成する必要がありますか?入力の各バッチの次元が100 * 10 * 5(バッチに使用される最初の次元)か10 * 100 * 5(バッチに2番目の次元が使用される)であるか?
以下の規則を説明する関連文書へのリンクがあります。
容器やモジュールのコンベンションは変更されますか?
おかげマヌエルを供給する前に「準備」する必要がありますどのように表示されます - https://github.com/Element-Research/rnn#rnn.Sequencerここでは、2番目の次元はデフォルトでバッチに使用されます。私は、異なるコンテナが違うコンベンションを使用していると思います。 –
@VikramGuptaあなたは正しいです!この場合、シーケンサーのコンテナは異なる入力シェイプを必要とするようです。 –