2016-07-13 5 views
0

トーチでバッチ処理に使用する入力の次元はどれですか?トーチモジュールに供給する前に入力をバッチする必要がありますか?

私はトレーニングのための1000の例があり、各トレーニングの例は10 * 5の次元です。さて、私は100の例のバッチとしてこのデータをSequencerに送りたいと思っています。

入力はどのように構成する必要がありますか?入力の各バッチの次元が100 * 10 * 5(バッチに使用される最初の次元)か10 * 100 * 5(バッチに2番目の次元が使用される)であるか?

以下の規則を説明する関連文書へのリンクがあります。

容器やモジュールのコンベンションは変更されますか?

答えて

0

通常は、サイズ100*10*5のテンソルです。画像の場合は、チャンネル数を考慮してbatchSize*channels*width*heightにする必要があります。これによりデータにアクセスしやすくなります。データを取得するにはinputs[{i}]を実行するだけです。ラベルを保存する別のTensorを作成することを検討します(ラベル付きデータを使用する場合)。ここでその例を見つけることができます。 https://github.com/torch/tutorials/blob/master/2_supervised/4_train.lua#L131

私はチュートリアルを見てすることをお勧めいたしますたい、そこは実際に私はこれを見た後に混乱しています、データがネットワークhttps://github.com/torch/tutorials

+1

おかげマヌエルを供給する前に「準備」する必要がありますどのように表示されます - https://github.com/Element-Research/rnn#rnn.​​Sequencerここでは、2番目の次元はデフォルトでバッチに使用されます。私は、異なるコンテナが違うコンベンションを使用していると思います。 –

+0

@VikramGuptaあなたは正しいです!この場合、シーケンサーのコンテナは異なる入力シェイプを必要とするようです。 –

関連する問題