numpy.apply_along_axis

今天看到一个numpy的函数，对一个tensor沿着某一个axis的slice分别应用某一个函数：

一定要注意那个axis，沿着某一个轴应用fun，比如第一个函数，是沿着【0】，还是【1】进行计算的

两个例子：

>>> def my_func(a):
... """Average first and last element of a 1-D array"""
... return (a[0] + a[-1]) * 0.5
>>> b = np.array([[1,2,3], [4,5,6], [7,8,9]])
>>> np.apply_along_axis(my_func, 0, b)
array([ 4., 5., 6.])
>>> np.apply_along_axis(my_func, 1, b)
array([ 2., 5., 8.])

>>> b = np.array([[8,1,7], [4,3,9], [5,2,6]])
>>> np.apply_along_axis(sorted, 1, b)
array([[1, 7, 8],
[3, 4, 9],
[2, 5, 6]])