def geometry

stelliferous@0.0.2.0 · tensor_band_concat.bend:11

import Base
import stelliferous@0.0.2.0/tensor_band.bend as Bands

11def geometry(shape: List<&2,U32>,axis: U32) -> Maybe<&2,Bands.Geometry>:
12  match shape axis:
13    case [channels,height,width] 0: Some{Bands.Geometry{1,channels,height,width}}
14    case [batch,channels,height,width] 1: Some{Bands.Geometry{batch,channels,height,width}}
15    case shape axis: None{}