def dropped_next

bendtorch-core@0.1.0.0 · src/autograd/sum_dims.bend:48

import Base

48def dropped_next(reduced: Bool, extent: U32, rest: List<&2, U32>) -> List<&2, U32>:
49  match reduced:
50    case True{}:
51      rest
52    case False{}:
53      extent <> rest