std::tuple statistics_cpu()

in src/inplace_abn_cpu.cpp [93:105]

• 10 lines of code
• 1 McCabe index (conditional complexity)

std::tuple<at::Tensor, at::Tensor, at::Tensor> statistics_cpu(
    const at::Tensor& x_) {
  CHECK_NOT_HALF(x_);

  auto x = normalize_shape(x_);

  auto mean = x.mean(c10::IntArrayRef({0, 2}));
  auto var = (x - normalize_shape(mean)).pow(2).mean(c10::IntArrayRef({0, 2}));
  auto count =
      at::full({1}, count_samples(x), x.options().dtype(at::ScalarType::Long));

  return std::make_tuple(mean, var, count);
}