def __init__(self, data, volatile=flag.OFF, name=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 = None
self.creator = None
self.name = name
评论列表
文章目录