torch.optim.rmsprop.rmsprop

torch.optim.rmsprop.rmsprop(params, grads, square_avgs, grad_avgs, momentum_buffer_list, state_steps, foreach=None, maximize=False, differentiable=False, capturable=False, has_complex=False, *, lr, alpha, eps, weight_decay, momentum, centered)

执行 RMSprop 算法计算的函数式 API。

详情请参阅 RMSProp。