将函数映射到多维数组的数组元素上

Question

背景：

• 我有一个函数 hlpr_fun ，它接受两个参数 a 和 b 并返回一个float列表。参数 a 是一个大小为12x2的二维数组， b 是一个由随机数组成的相同大小的数组。 hlpr_fun 从内部调用另一个函数 grad_calc 。

  freq[4]=5

问题是这样的：

• 我需要为 a 和 b 的100,000个不同值调用函数 hlpr_fun 100,000次。所以，我所做的是首先创建大小为100000x12x2的数组 a_array 和 b_array 。我正在尝试np.vectorize，但有一些错误。

是否有一种有效的方法将函数 hlpr_fun 映射到沿着轴= 0传播的 a_array 和 b_array 的每个12x2阵列对中？