例如,我有一个形状如下的矩阵: x = np.random.rand(3, 10, 2, 6) 如您所见,一个axis=2上只有两个数组。我有一个接受这两个数组的函数: def f(arr1, arr2): # arr1 with shape (6, ) and arr2 with (6, )
return np.sum(arr1, arr2) # for simplicity 如何将该函数沿第二轴以矢量化的方式应用于x数组?使得所得到阵列将具有[3, 10, dim of output]形状。