implement Variable.xp

chainer · Mar 19, 2018 · ba6c412 · ba6c412
1 parent e788405
commit ba6c412
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/chainer/variable.py b/chainer/variable.py
@@ -487,6 +487,16 @@ def __repr__(self):
     def __str__(self):
         return variable_str(self)
 
+    @property
+    def xp(self):
+        """Array module for this variable.
+
+        Depending on which of CPU/GPU this variable is on, this property
+        returns :mod:`numpy` or :mod:`cupy`.
+
+        """
+        return cuda.get_array_module(self)
+
     @property
     def name(self):
         return self._node.name