Я хотел бы знать, как использовать API-интерфейс tenorflow C ++: tensorflow::ops::SaveV2
для создания контрольных точек. Его подпись:
SaveV2(
const ::tensorflow::Scope & scope,
::tensorflow::Input prefix,
::tensorflow::Input tensor_names,
::tensorflow::Input shape_and_slices,
::tensorflow::InputList tensors
)
У меня есть префикс в виде строки c ++, различные тензоры (Tensor
объекты), которые я хотел бы сохранить, включая их имена (std::vector<string> names
) и формы (TensorShape
объектов). Моя проблема в том, как привести их к разным типам в подписи. У меня нет проблем с параметром области. Моя проблема заключается в префиксе, тензорных именах, shape_and_slices и тензорах. Я буду рад, если кто-нибудь сможет написать пример блока кода, реализующего упомянутую подпрограмму с перечисленными типами.