Merge pull request #144 from mattzque/pytorch-1.3.0

Support for pytorch 1.3.0 added.
mapillary · Nov 22, 2019 · 1b91300 · 1b91300
2 parents fdd8bd9 + 93b8588
commit 1b91300
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/inplace_abn_cpu.cpp b/src/inplace_abn_cpu.cpp
@@ -73,8 +73,8 @@ std::tuple<at::Tensor, at::Tensor, at::Tensor> statistics_cpu(const at::Tensor&
 
   auto x = normalize_shape(x_);
 
-  auto mean = x.mean({0, 2});
-  auto var = (x - normalize_shape(mean)).pow(2).mean({0, 2});
+  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);