mindspore.ConvertNetUtils

class mindspore.ConvertNetUtils[源代码]

将网络转换为thor层网络,用于计算并存储二阶信息矩阵。

convert_to_thor_net(net)[源代码]

该接口用于将网络转换为thor层网络,用于计算并存储二阶信息矩阵。

Note

此接口由二阶优化器thor自动调用。

参数:
  • net (Cell) - 由二阶优化器thor训练的网络。

支持平台:

Ascend GPU

样例:

>>> ConvertNetUtils().convert_to_thor_net(net)