torch::Tensor out = torch::randn({100, 12}).toType(torch::kFloat32);
std::cout << torch::mean(out, 0).sizes() << "\n";
std::cout << torch::var(out, 0).sizes() << "\n";

[12]
[]


Anyone know whats wrong?