def sum_axis

stelliferous@0.0.2.0 · tensor_f32.bend:365

import Base
import stelliferous@0.0.2.0/tensor_layout.bend as Layout

365# Reductions along one axis remove that axis. Each result element takes the
366# elements of the axis in increasing index, starting from 0, -inf or +inf.
367def sum_axis(tensor: Tensor(),+axis: U32) -> Tensor():
368  Layout.reduce_axis(~F32,~F32.add,"sum_axis",0.0,tensor,axis)