def repeated_axes

stelliferous@0.0.2.0 · tensor_layout.bend:141

import Base
import stelliferous@0.0.2.0/tensor_view.bend as Views

141def repeated_axes(axes: List<&2,Views.Axis>,+scale: U32) -> Maybe<&2,List<&2,Views.Axis>>:
142  match axes:
143    case [Views.Axis{height,height_stride},Views.Axis{width,width_stride}]:
144      Some{[Views.Axis{height,height_stride},Views.Axis{scale,0},Views.Axis{width,width_stride},Views.Axis{scale,0}]}
145    case [axis] : None{}
146    case []: None{}
147    case axis <> rest: prepend_repeated(axis,repeated_axes(rest,scale))