def dense_rows

stelliferous@0.0.2.0 · tensor_band_concat.bend:83

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

83def dense_rows(~Element: Data,geometry: Bands.Geometry,+first: U32,+rows: U32,channel: U32,input: Storage.Buffer<Element>,output: Storage.Buffer<Element>) -> Storage.Buffer<Element> & Storage.Buffer<Element>:
84  Bands.Geometry{batch,+channels,+height,+width} = geometry
85  Bands.planes(~Element,U32.to_nat(channels),0,(rows * width : U32),(height * width : U32),(rows * width : U32),
86    ((first / height * channels * height + first % height) * width : U32),(channel * rows * width : U32),(input,output))