def reorder_head

bendtorch-core@0.1.0.0 · src/tensor/layout.bend:54

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

import Base

54def reorder_head(rec: U32 -> Maybe<&2, List<&2, U32>>, head: Maybe<&2, U32>) -> Maybe<&2, List<&2, U32>>:
55  match head:
56    case None{}:
57      None{}
58    case Some{value}:
59      rec(value)

Uses (3)

Used by (5)