Theano在某些维度上添加张量

时间:2016-11-19 00:55:48

标签: theano numpy-broadcasting

我试图了解如何使用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中这样做。

1 个答案:

答案 0 :(得分:1)

使用dimshuffle

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