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
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
def volatile(self, v): self._volatile = flag.Flag(v)