def input_shape

stelliferous@0.0.2.0 · tensor_convolution.bend:33

import Base

33def input_shape(shape: List<&2,U32>) -> Maybe<&2,InputShape>:
34  match shape:
35    case [channels,height,width]: Some{InputShape{1,False{},channels,height,width}}
36    case [batch,channels,height,width]: Some{InputShape{batch,True{},channels,height,width}}
37    case shape: None{}