eazygrad.clip

eazygrad.clip(input: _Tensor, low: float, high: float) _Tensor[source]

Clip tensor values to a closed interval.

Parameters:
  • input (_Tensor) – Input tensor.

  • low (scalar) – Lower clipping bound.

  • high (scalar) – Upper clipping bound.

Returns:

_Tensor – Clipped tensor.

See also

torch.clamp