commit e2fad34067f57b04b8d67b4b36528e6018a0df4f
author Ashutosh Hathidara <ashutosh.hathidara@legatohealth.com> Fri May 28 21:32:36 2021 +0530
committer Ashutosh Hathidara <ashutosh.hathidara@legatohealth.com> Fri May 28 21:32:36 2021 +0530
tree 9dbed2eedc6fdb7d07d1bde0392a0e72b3403734
parent 82740e347758113bf703b69b262bb2de3dc7b2ea

Added axis

tensorflow/python/ops/math_ops_test.py

diff --git a/tensorflow/python/ops/math_ops_test.py b/tensorflow/python/ops/math_ops_test.py
index ce2d7e9..11dbfd0 100644
--- a/tensorflow/python/ops/math_ops_test.py
+++ b/tensorflow/python/ops/math_ops_test.py
@@ -100,7 +100,8 @@
       [5., 9., 2.],
       [5.],
       []])
-    self.assertAllClose(math_ops.reduce_variance(x), [0., 16., 1., 0.])
+    self.assertAllClose(math_ops.reduce_variance(x, axis=0),
+      [0., 16., 1., 0.])
 
   def testReduceVarComplex(self):
     # Ensure that complex values are handled to be consistent with numpy