def include_axis

stelliferous@0.0.2.0 · tensor_cursor.bend:12

import stelliferous@0.0.2.0/tensor_cursor.bend as TensorCursor

import Base

12def include_axis(active: Bool,+extent: U32,+stride: U32,axes: List<&2,Axis>) -> List<&2,Axis>:
13  match active:
14    case False{}: axes
15    case True{}:
16      +count = U32.to_nat(extent)
17      Axis{count,count,stride,((extent - 1) * stride : U32)} <> axes

Uses (7)

Used by (1)

stelliferous@0.0.2.0/tensor_cursor:axes