Pythorch中的批矩阵乘法与输出维数的处理相混淆

Question

我有两个阵列：

A
B

数组A包含一批RGB图像，形状：

^{pr2}$

而数组B包含对图像进行“类似变换”操作所需的系数，形状：

[batch, 4, 4, 3]

简单地说，单个图像的操作是输出环境映射（normalMap * Coefficients）的乘法。在

我想要的输出应该保持形状：

^{pr2}$

我尝试使用torch.bmm，但失败了。这有可能吗？在

Answer 1

我想你需要计算一下Pythorch的工作原理

BxCxHxW : number of mini-batches, channels, height, width

格式，并且还使用^{}，因为^{}与3D张量一起工作。在

我知道你可能会在网上找到这个，但无论如何：

^{pr2}$