python - Numpy:我可以在没有 einsum 的情况下进行张量 Hadamard 乘法吗？

Question

我有一个四阶张量 A(比如索引 (a, b, i, j))和一个二阶张量 B(比如索引 (i, j))，我想计算一种 Hadamard 乘法他们中的。

也就是说，如果我们称产品为 C，我想要 C[a,b,i,j] == A[a,b,i,j] * B[i,j] .使用 einsum 有一种相当简单的方法可以做到这一点但有人告诉我使用 einsum 会严重影响性能与 tensordot 等方法相比，但我找不到从文档中避免它的好方法。

有可能我错过了；我是张量的新手，而不是 numpy 的专家。

Answer 1

C = A * B

在 broadcasting rules 之后, NumPy 将从最后一个轴开始排列 A 和 B 的形状:

A: (a, b, i, j)
B:       (i, j)

并将 A 和 B 的相应元素相乘以创建 C:

   A[a, b, i, j]
*  B[      i, j]
== C[a, b, i, j]