Esempio n. 1
0
    def __init__(self, data, volatile=flag.OFF, name=None):
        assert isinstance(data, (numpy.ndarray, cuda.ndarray))

        self.data = data
        self.rank = 0
        self._volatile = flag.Flag(volatile)

        self._grad = None
        self.creator = None

        self.name = name
Esempio n. 2
0
    def __init__(self, data, volatile=flag.OFF, name=None, grad=None):
        if not isinstance(data, (numpy.ndarray, cuda.ndarray)):
            msg = '''numpy.ndarray or cuda.ndarray are expected.
Actual: {0}'''.format(type(data))
            raise TypeError(msg)

        self.data = data
        self.rank = 0
        self._volatile = flag.Flag(volatile)

        self._grad = grad
        self.creator = None

        self.name = name
Esempio n. 3
0
 def volatile(self, v):
     self._volatile = flag.Flag(v)