mlx.core.stop_gradient

stop_gradient(a: array, /, *, stream: None | Stream | Device = None) → array

阻止计算梯度。

该操作本身是恒等操作，但它阻止梯度流经此数组。

参数:

a (array) – 输入数组。

返回值:

未更改的输入 a，但阻止梯度流经它。

返回类型:

array