def reorder_step

bendtorch-core@0.1.0.0 · src/tensor/axis_algebra.bend:21

import Base
import bendtorch-core@0.1.0.0/src/tensor/layout.bend as Layout
import bendtorch-core@0.1.0.0/src/tensor/axis_spec.bend as Axis

21def reorder_step(+head: Maybe<&2, U32>, +tail: Maybe<&2, List<&2, U32>>) ->
22  {Layout.reorder_head(value => Layout.reorder_tail(value, tail), head) == Axis.prepend(U32, head, tail) : Maybe<&2, List<&2, U32>>}:
23  match head tail:
24    case None{} None{}:
25      {==}
26    case None{} Some{items}:
27      {==}
28    case Some{value} None{}:
29      {==}
30    case Some{value} Some{items}:
31      {==}