def dropped_from

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

import Base

55# The dimensions from axis `axis` on, without the reduced ones.
56def dropped_from(+axes: List<&2, U32>, dimensions: List<&2, U32>, +axis: U32) -> List<&2, U32>:
57  match dimensions:
58    case Nil{}:
59      Nil{}
60    case extent <> rest:
61      dropped_next(reduces(axes, axis), extent, dropped_from(axes, rest, U32.add(axis, 1)))