cupy.apply_along_axis#
- cupy.apply_along_axis(func1d, axis, arr, *args, **kwargs)[source]#
沿给定轴对一维切片应用函数。
- 参数：
func1d (function (M,) -> (Nj...)) – 此函数应接受一维数组。它沿指定的轴应用于
arr的一维切片。它必须返回一个一维cupy.ndarray。axis (integer) – 用于对
arr进行切片的轴。arr (cupy.ndarray (Ni..., M, Nk...)) – 输入数组。
args –
func1d的附加参数。kwargs –
func1d的附加关键字参数。
- 返回值：
输出数组。
out的形状与arr的形状相同，但沿axis维度除外。该轴被移除，并替换为等于func1d返回值形状的新维度。因此，如果func1d返回一个标量，则out将比arr少一个维度。- 返回类型：