def choice_output

bendtorch-core@0.1.0.0 · src/shape/plan_proofs.bend:58

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

58def choice_output(-left: U32, -right: U32, choice: Shape.AxisChoice<left, right>) -> Maybe<&2, Nat>:
59  match choice:
60    case Shape.Picks{output, axis}:
61      Some{U32.to_nat(output)}
62    case Shape.Clashes{}:
63      None{}