def max_axis

stelliferous@0.0.2.0 · tensor_f32.bend:370

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

370def max_axis(tensor: Tensor(),+axis: U32) -> Tensor():
371  Layout.reduce_axis(~F32,~F32.max,"max_axis",F32.div(F32.neg(1.0),0.0),tensor,axis)