Theanoは特定の次元に沿ってテンソルを追加します

Question

dimshuffleを使用してノイズを4次元テンソルにブロードキャストする方法を理解しようとしています。例えば、4次元テンソルは（num_examples、3,32,32）であり、3は32 * 32画像のr、g、b成分に対応する。私はガウスノイズを追加したいが、3つのコンポーネントすべてに同じノイズ、すなわちr、g、bを追加したいとし、それぞれの例でノイズはランダムな32 * 32のマトリックスになる。だから私は次元の2つのテンソル（num_examples、3,32,32）と（num_examples、32,32）をそれぞれ追加しなければなりません。私はそれの最後の2つの次元がブロードキャスト可能になるように、第2のテンソルでdimshuffleを使用する方法を理解することができませんでした。

私はtheanoでこれをしたいと思います。答えを

Answer 1

使用dimshuffle

x = T.tensor4() # Nx3x32x32 
y = T.tensor3() # Nx32x32 
z = y.dimshuffle(0, 'x', 1, 2) + x