def _get_adam_feature(self, learning_rate, m, v):
    epsilon = 1e-08
    return torch.mul(learning_rate, (m / (torch.sqrt(v) + epsilon)))
def _get_adam_feature(self, learning_rate, m, v):
    epsilon = 1e-08
    return torch.mul(learning_rate, (m / (torch.sqrt(v) + epsilon)))