如何向量化传递一个函数到两个numpy数组: 3D和2D?

我有两个多维numpy数组:x是3D的,y是2D的。

如果我有一个函数foo(a, b),它接受两个二维数组作为输入,我如何传递给foo我的多维数组,并以向量化的方式遍历x的3维,以获得foo的结果列表?

我一直在尝试使用np.vectorize来做这件事,但是它遍历数组的各行并产生错误,所以我被卡住了。

转载请注明出处:http://www.intsu.net/article/20230526/1073523.html