From 506840b45d449727d231726dcae61fbf04567c7a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Bart=20van=20Merri=C3=ABnboer?= Date: Mon, 24 Jul 2017 15:38:33 -0400 Subject: [PATCH] Add vspace support for cupy --- autograd/cupy/cupy_extra.py | 3 +++ autograd/util.py | 1 + 2 files changed, 4 insertions(+) diff --git a/autograd/cupy/cupy_extra.py b/autograd/cupy/cupy_extra.py index 68b415d3..a817862a 100644 --- 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 index 613aebaa..06d6b62f 100644 --- 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