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.