mindscience.common.AdaHessian ============================== .. py:class:: mindscience.common.AdaHessian 二阶优化器 AdaHessian,利用 Hessian 矩阵对角元信息进行二阶优化求解。 有关更多详细信息,请参考论文 `ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning `_ 。 此处 Hessian power 固定为 ``1``,且对 Hessian 对角元做空间平均的方法如下: - 对于 1D 张量:不做空间平均。 - 对于 2D 张量:做行平均。 - 对于 3D 张量(假设为 1D 卷积):对最后一个维度做平均。 - 对于 4D 张量(假设为 2D 卷积):对最后两个维度做平均。 参数说明详见 `mindspore.nn.Adam `_ 。