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