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