stop_gradient¶

dragon.stop_gradient(
  inputs,
  **kwargs
)[source]¶

Return the identity of input with truncated gradient-flow.

The expression itself is unaffected, but gradient is stopped.

Parameters:

• inputs (dragon.Tensor) – The input tensor.

Returns:

dragon.Tensor – An identity of input.