def mean

stelliferous@0.0.2.0 · tensor_f32.bend:310

import Base
import stelliferous@0.0.2.0/tensor.bend as Tensors

310# sum divided by the number of elements; an empty tensor gives NaN.
311def mean(tensor: Tensor()) -> Tensor():
312  then_shape(shape(tensor),tensor => extents =>
313    Tensors.combine_scalar(~F32,~F32.div,"mean",1.0,Tensors.reduce_all(~F32,~F32.add,"mean",tensor,0.0),
314      U32.to_f32(Tensors.count_or_zero(extents))))