cupyx.scipy.ndimage.generic_gradient_magnitude#

cupyx.scipy.ndimage.generic_gradient_magnitude(input, derivative, output=None, mode='reflect', cval=0.0, extra_arguments=(), extra_keywords=None)[源码]#

使用提供的导数函数进行多维梯度幅值滤波。

参数:
  • input (cupy.ndarray) – 输入数组。

  • derivative (callable) –

    具有以下签名的函数或其他可调用对象,每个轴调用一次

    derivative(input, axis, output, mode, cval,
               *extra_arguments, **extra_keywords)
    

    其中 inputoutputcupy.ndarrayaxis 是从 0 到维数之间的 intmodecvalextra_argumentsextra_keywords 是传递给此函数的值。

  • output (cupy.ndarray, dtypeNone) – 放置输出结果的数组。默认与输入的 dtype 相同。

  • mode (str) – 数组边界根据给定模式('reflect''constant''nearest''mirror''wrap')处理。默认是 'reflect'

  • cval (标量) – 如果模式是 'constant',用于填充输入边缘之外区域的值。默认是 0.0

  • extra_arguments (序列, 可选) – 传递给 derivative2 的额外位置参数序列。

  • extra_keywords (dict, 可选) – 传递给 derivative2 的额外关键字参数字典。

返回值:

滤波的结果。

返回值类型:

cupy.ndarray

注意

当输出数据类型为整型时(或者在未提供输出且输入为整型时),由于中间结果的浮点舍入,结果可能与 SciPy 的结果不完全匹配。