def stretch_axes

stelliferous@0.0.2.0 · tensor_view.bend:224

import Base

224def stretch_axes(axes: List<&2,Axis>,target: List<&2,U32>) -> Maybe<&2,List<&2,Axis>>:
225  match axes target:
226    case Nil{} Nil{}: Some{[]}
227    case Con{Axis{+size,stride},tail} Con{+extent,rest}:
228      stretched_prepend(stretched_axis(U32.is_eq(size,extent),U32.is_eq(size,1),size,stride,extent),stretch_axes(tail,rest))
229    case axes target: None{}