实现 Adam 优化器更新步骤函数。您的函数应接受当前参数值、梯度以及移动平均作为输入,并返回更新后的参数值和新移动平均。该函数还应处理标量和数组输入,并包括对移动平均的偏差校正。
输入描述:
第一行输入五个浮点数,分别代表参数、梯度、一阶矩估计、二阶矩估计、迭代次数。


输出描述:
输出更新后的参数、一阶矩估计、二阶矩估计。结果都保留五位小数。
示例1

输入

1.0
0.1
0.0
0.0
1

输出

0.999 0.01 1e-05
示例2

输入

[1.0,2.0]
[0.1,0.2]
[0,0]
[0,0]
1

输出

[0.999 1.999] [0.01 0.02] [1.e-05 4.e-05]
加载中...