def split

stelliferous@0.0.2.0 · tensor_band.bend:391

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

391# Plan metadata first, then transfer each dense channel range once. First
392# splits and later changed boundaries share the same copy implementation.
393def split(~Element: Data,depth: Nat,+geometry: Geometry,fill: Element,input: Storage.Buffer<Element>) -> Bands(~Element):
394  Geometry{batch,channels,height,width} = geometry
395  from_dense(~Element,geometry,fill,image_plan(U32.to_nat(batch),0,depth,height),input)