Add vspace support for cupy

HIPS · Jul 31, 2017 · 506840b · 506840b
1 parent 48c7096
commit 506840b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/autograd/cupy/cupy_extra.py b/autograd/cupy/cupy_extra.py
@@ -82,6 +82,9 @@ def __init__(self, value):
     def zeros(self):
         return acp.zeros(self.shape, dtype=self.dtype)
 
+    def ones(self):
+        return acp.ones(self.shape, dtype=self.dtype)
+
     def flatten(self, value, covector=False):
         return acp.ravel(value)
 

diff --git a/autograd/util.py b/autograd/util.py
@@ -4,6 +4,7 @@
 from builtins import range
 
 import autograd.numpy as np
+import autograd.cupy as cp
 from autograd.convenience_wrappers import grad
 from autograd.core import vspace, vspace_flatten, getval