Function MSModelGetWeights

Function Documentation

MSTensorHandleArray MSModelGetWeights(MSModelHandle model)

Obtains all weights tensors of the model. Only valid for Lite Train.

Parameters:

model – [in] Model object handle.

Returns:

The vector that includes all gradient tensors.