def dropped

bendtorch-core@0.1.0.0 · src/tensor/slice.bend:132

import Base

132# The input dimensions without the axis, as `select` returns them.
133def dropped(dimensions: List<&2, U32>, +dim: Nat) -> List<&2, U32>:
134  match dimensions dim:
135    case head <> rest 0n:
136      rest
137    case head <> rest 1n+next:
138      head <> dropped(rest, next)
139    case Nil{} _:
140      Nil{}