python - 使用 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 以便其最后两个维度变得可广播。

Answer 1

使用dimshuffle

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