Эта функция предоставляет способ импортировать сериализованный протокол буфера 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 не является корректным (например, он ссылается на неизвестный тензор).