Как использовать tenorflow C ++ API tenensflow :: ops :: SaveV2 для создания контрольных точек при обучении модели в C ++ - PullRequest
0 голосов
/ 09 ноября 2019

Я хотел бы знать, как использовать 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 и тензорах. Я буду рад, если кто-нибудь сможет написать пример блока кода, реализующего упомянутую подпрограмму с перечисленными типами.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...