举一个函数的一个小例子,这里有两个数组,我们想要逐个元素相乘,然后沿轴1(数组的行)求和:
A= np.array([0,1,2])
B= np.array([[0, 1, 2, 3],...首先要注意的是我们需要reshapeA,这样我们在乘B时才可以广播(就是说,A需要是列向量)。然后我们可以用B的第一行乘以0,第二行乘以1,第三行乘以2。...我们要相乘的两个数组是:
A= np.array([[1,1,1],
[2,2,2],
[5,5,5]])
B= np.array([[0,1,0...要了解输出数组的计算方法,请记住以下三个规则:
在输入数组中重复的字母意味着值沿这些轴相乘。乘积结果为输出数组的值。
在本例中,我们使用字母j两次:A和B各一次。这意味着我们将A每一行与B每列相乘。...这只在标记为j的轴在两个数组中的长度相同(或者任一数组长度为1)时才有效。
输出中省略的字母意味着沿该轴的值将相加。
在这里,j不包含在输出数组的标签中。