Numpy：シェイプのndarray（...、3）をf（（shape（3）ndarray） - >（float）にマッピングすると、ndarray（...）が得られますか？

Question

私はshape（3）のndarrayをfloatにマップし、ndarray（shape）（...、3）を持つ関数を持っています。その配列にその関数をマップして形状（...）の配列を取得する最良の方法は何ですか？

ありがとうございました。

Answer 1

希望するnumpy.apply_along_axis

def f(a): 
    return a[0] + a[1] + a[2] 
mm = numpy.random.randn(5, 3) 
numpy.apply_along_axis(f, 1, mm) 

出力：array([-1.75875289, -0.34689792, 0.66092486, -0.21626001, -0.14125476])