Theano Hinzufügen von Tensoren in bestimmten Dimensionen

Question

Ich versuche zu verstehen, wie DimScuffle verwendet wird, um das Rauschen auf den 4-dimensionalen Tensor zu übertragen. Zum Beispiel ist der 4-dimensionale Tensor (num_examples, 3,32,32), wobei 3 den r, g, b Komponenten des 32 * 32 Bildes entspricht. Ich möchte Gaußsches Rauschen hinzufügen, aber ich möchte allen drei Komponenten, d. H. R, g, b, das gleiche Rauschen hinzufügen, und für jedes Beispiel wäre das Rauschen eine zufällige 32 * 32-Matrix. Also muss ich die beiden Tensoren der Dimensionen (num_examples, 3,32,32) und (num_examples, 32,32) hinzufügen. Ich bin nicht in der Lage, herauszufinden, wie man Dim-Shuffle auf dem zweiten Tensor verwendet, so dass die letzten beiden Dimensionen davon übertragbar werden.

Ich möchte dies in Theano tun.

Answer 1

Verwenden dimshuffle

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