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