fix inplace division for python3 (pytorch#2063)

moyamao · Jul 12, 2017 · a74fb22 · a74fb22
1 parent 0d91048
commit a74fb22
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/test/test_torch.py b/test/test_torch.py
@@ -4054,6 +4054,13 @@ def test_big_transpose(self):
         t2 = torch.from_numpy(t.numpy().transpose())
         self.assertEqual(t1, t2)
 
+    def test_inplace_division(self):
+        t = torch.rand(5, 5)
+        id_before = id(t)
+        t /= 2
+        id_after = id(t)
+        self.assertEqual(id_before, id_after)
+
 # Functions to test negative dimension wrapping
 METHOD = 1
 INPLACE_METHOD = 2

diff --git a/torch/tensor.py b/torch/tensor.py
@@ -333,6 +333,7 @@ def __rdiv__(self, other):
 
     def __idiv__(self, other):
         return self.div_(other)
+    __itruediv__ = __idiv__
 
     def __mod__(self, other):
         return self.remainder(other)