def fold_slices

stelliferous@0.0.2.0 · tensor_layout.bend:430

import Base
import stelliferous@0.0.2.0/storage_buffer.bend as Storage

430def fold_slices(~Element: Data,~operation: Element -> Element -> Element,remaining: Nat,+index: U32,+axis: U32,+shape: List<&2,U32>,+unit: List<&2,U32>,
431  state: (Storage.Buffer<Element> & Storage.Buffer<Element>) & Bool) -> (Storage.Buffer<Element> & Storage.Buffer<Element>) & Bool:
432  match remaining:
433    case 0n: state
434    case 1n+rest:
435      fold_slices(~Element,~operation,rest,(index + 1 : U32),axis,shape,unit,fold_step(~Element,~operation,index,axis,shape,unit,state))