def build_axes

stelliferous@0.0.2.0 · tensor_view.bend:70

import stelliferous@0.0.2.0/tensor_view.bend as TensorView
import Base
import stelliferous@0.0.2.0/tensor_shape.bend as Shapes

70def build_axes(reversed: List<&2,U32>,stride: Maybe<&2,U32>,axes: List<&2,Axis>,offset: U32) -> Maybe<&2,View>:
71  match reversed stride:
72    case Nil{} Some{value}: Some{View{axes,offset}}
73    case Con{+size,tail} Some{+value}:
74      build_axes(tail,Shapes.checked_multiply(value,size),Axis{size,value} <> axes,offset)
75    case reversed None{}: None{}

Uses (6)

Used by (1)