def gather_leaf

stelliferous@0.0.2.0 · tensor_band.bend:55

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

55def gather_leaf(~Element: Data,geometry: Geometry,+first: U32,+rows: U32,input: Storage.Buffer<Element>,output: Storage.Buffer<Element>) -> Storage.Buffer<Element>:
56  Geometry{batch,+channels,+height,+width} = geometry
57  destination(~Element,planes(~Element,U32.to_nat(channels),0,(rows * width : U32),(rows * width : U32),(height * width : U32),0,
58    ((first / height * channels * height + first % height) * width : U32),(input,output)))