mean method

Tensor mean(
  1. int dim, {
  2. bool keepDim = false,
})

Mean over a given dimension.

Implementation

Tensor mean(int dim, {bool keepDim = false}) {
  final s = sum(dim, keepDim: keepDim);
  return s.divScalar(shape[dim].toDouble());
}