

def init_state(self, param, state):
    xp = cuda.get_array_module(param.data)
    state['ms'] = xp.zeros_like(param.data)

dataset/ETHPy150Open pfnet/chainer/chainer/optimizers/rmsprop.py RMSprop.init_state/original