mindspore.ConvertNetUtils

class mindspore.ConvertNetUtils[source]

Convert net to thor layer net, used to compute and store second-order information matrix.

convert_to_thor_net(net)[source]

This interface is used to convert a network to thor layer network, in order to calculate and store the second-order information matrix.

Note

This interface is automatically called by the second-order optimizer thor.

Parameters: net (Cell) – Network to be trained by the second-order optimizer thor.

Examples

>>> ConvertNetUtils().convert_to_thor_net(net)