def pixel
stelliferous@0.0.2.0 · kernel_packing_samples.bend:16
import stelliferous@0.0.2.0/kernel_packing_samples.bend as KernelPackingSamples
import Base
16def pixel(input_storage: Array<F32>, +output_position: U32, +reduction_index: U32, +input_height: U32, +input_width: U32, +kernel_size: U32, +stride: U32, +padding: U32, +output_width: U32) -> Array<F32> & F32:
17 +input_row = ((output_position / output_width) * stride + (reduction_index / kernel_size) % kernel_size - padding : U32)
18 +input_column = ((output_position % output_width) * stride + reduction_index % kernel_size - padding : U32)
19 at = ((reduction_index / (kernel_size * kernel_size)) * input_height * input_width + input_row * input_width + input_column : U32)
20 mask_pixel(U32.is_lt(input_row, input_height) && U32.is_lt(input_column, input_width), Array.get(F32, input_storage, at))