def single_below

bendtorch-core@0.1.0.0 · src/autograd/reduce.bend:96

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape

96# `single_rows` of the rows below the first one.
97def single_below(+rows: List<&2, Shape.Row>, +extents: List<&2, Nat>) -> Bool:
98  match rows extents:
99    case Shape.Row{stride, left, right} <> +rest +extent <> +others:
100      single_rows(rest, others)
101    case _ _:
102      False{}