Theano在某些維度上添加張量

Question

我想了解如何使用dimshuffle將噪聲廣播到4維張量。例如，4維張量是（num_examples，3,32,32），其中3對應於32 * 32圖像的r，g，b分量。我想添加高斯噪聲，但是我想爲所有三個分量添加相同的噪聲，即r，g，b，並且對於每個示例，噪聲將是隨機的32 * 32矩陣。所以我必須分別添加維度的兩個張量（num_examples，3,32,32）和（num_examples，32,32）。我還沒有弄清楚如何在第二張張上使用dimshuffle，以便它的最後兩個維度變得可以播放。

我想在theano中做到這一點。

Answer 1

使用dimshuffle

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