def clamp

bendtorch-core@0.1.0.0 · src/tensor/slice_LAWS.bend:124

import Base

124# PyTorch's clamp of a slice position to an axis of extent n: the position
125# if it is at most n, else n.
126def clamp(+position: Nat, +extent: Nat) -> Nat:
127  Bool.pick(Nat, Nat.is_le(position, extent), position, extent)