Эта функция позволяет импортировать сериализованный буфер протокола TensorFlow GraphDef и извлекать отдельные объекты из GraphDef в виде объектов tf.Tensor и tf.Operation. После извлечения эти объекты помещаются в текущий граф по умолчанию Graph. См. раздел tf.Graph.as_graph_def, чтобы узнать, как создать протокол GraphDef.
Аргументы
graph_def
Протокол GraphDef, содержащий операции для импорта в граф по умолчанию.
input_map
Словарь, сопоставляющий имена входных данных (в виде строк) в graph_def с объектами Tensor. Значения именованных входных тензоров в импортированном графе будут сопоставлены с соответствующими значениями Tensor.
return_elements
Список строк, содержащий имена операций в graph_def, которые будут возвращены как объекты Operation, и/или имена тензоров в graph_def, которые будут возвращены как объекты Tensor.
name
(Необязательно.) Префикс, который будет добавлен к именам в graph_def. Обратите внимание, что это не относится к именам импортированных функций. По умолчанию — "import".
op_dict
(Необязательно.) Устарело, не используйте.
producer_op_list
(Необязательно.) Протокол OpList со списком OpDef, использованных создателем графа (возможно, сокращенным). Если он указан, нераспознанные атрибуты операций в graph_def, имеющие значения по умолчанию согласно producer_op_list, будут удалены. Это позволит принимать некоторые дополнительные GraphDef, созданные более поздними бинарными файлами, более ранними бинарными файлами.
Возвращает
Список объектов Operation и/или Tensor из импортированного графа, соответствующих именам в return_elements, и None, если returns_elements имеет значение None.
Вызывает исключения
TypeError
Если graph_def не является протоколом GraphDef, input_map не является словарём, сопоставляющим строки объектам Tensor, или return_elements не является списком строк.
ValueError
Если input_map или return_elements содержит имена, отсутствующие в graph_def, либо graph_def имеет неправильный формат (например, ссылается на неизвестный тензор).