TensorF32.sum_axis: def of stelliferous@0.0.2.0
Import: import stelliferous@0.0.2.0/tensor_f32.bend as TensorF32
Lines 365-368 of stelliferous@0.0.2.0/tensor_f32.bend: https://jend.leeeet.dev/read.txt?path=stelliferous@0.0.2.0/tensor_f32.bend&start_line=365&end_line=368

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

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

Uses (5):
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=F32
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=F32.add
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=U32
  https://jend.leeeet.dev/definition.txt?path=stelliferous@0.0.2.0/tensor_f32.bend&name=Tensor
  https://jend.leeeet.dev/definition.txt?path=stelliferous@0.0.2.0/tensor_layout.bend&name=reduce_axis