

def test_rmsprop(backend_default):
    rms = RMSProp()
    param = np.random.rand(200, 128)
    param2 = copy.deepcopy(param)
    grad = (0.01 * np.random.rand(200, 128))
    grad2 = (grad / 128.0)
    states = [(0.01 * np.random.rand(200, 128))]
    state = states[0]
    decay = rms.decay_rate
    denom = (np.sqrt((((decay * state) + (np.square(grad2) * (1.0 - decay))) + rms.epsilon)) + rms.epsilon)
    backend_default[:] -= ((grad2 * rms.learning_rate) / denom)
    param_list = [((wrap(param), wrap(grad)), [wrap(states[0])])]
    compare_tensors(rms, param_list, param2, tol=1e-07)

dataset/ETHPy150Open NervanaSystems/neon/tests/test_optimizer.py test_rmsprop/VarMisuse@137/150 `param2`->`backend_default`