1

我试图了解如何使用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在第二个张量上使用，以便它的最后两个维度变得可广播。

4

1 回答 1

1

采用dimshuffle

x = T.tensor4() # Nx3x32x32
y = T.tensor3() # Nx32x32
z = y.dimshuffle(0, 'x', 1, 2) + x
于 2016-11-23T10:51:29.267 回答