手写 Adam 优化器单步更新

从零实现 Adam 优化器的一步参数更新。给定参数、梯度、一阶/二阶动量状态和超参数，执行带偏差修正的 Adam 更新，返回更新后的参数： m' = beta1 * m + (1 - beta1) * grad v' = beta2 * v + (1 - beta2) * grad ** 2 m_hat = m' / (1 - beta1 ** t) # 偏差修正 v_hat = v' / (1 - beta2 ** t) param' = param - lr * m_hat / (sqrt(v_hat) + eps) 返回: 更新后的参数张量（形状与输入 param 相同）。 要求： - 不要调用 torch.optim.Adam / torch.optim._functional； - 严格按上述公式顺序实现（先更新动量，再做偏差修正）； - 不要原地修改传入的 param/m/v 张量。