Numpy：用max代替sum的点积

Question

在numpy中有没有一种方法可以做到以下几点（或者有一个通用的数学术语）：

假设正常点积：

M3[i,k] = sum_j(M1[i,j] * M2[j,k])

现在我想用sum替换sum其他操作，比如最大值：

^{pr2}$

正如你所看到的，它与上面的完全平行，我们只取max，而不是总和。在

其他选项可以是min，prod，以及任何其他将序列/集转换为值的操作。在

Answer 1

正常的点积应该是（使用numpy广播）

M3 = np.sum(M1[:, :, None] * M2[None, :, :], axis = 1)

您可以对任何具有axis关键字的函数执行相同的操作。在

M3 = np.max(M1[:, :, None] * M2[None, :, :], axis = 1)