Spec-Zone.ru › TensorFlow 1.15

tf.io.write_graph

Просмотреть исходный код на GitHub

Записывает протокол графа в файл.

Просмотр псевдонимов

Основные псевдонимы

`tf.train.write_graph`

Псевдонимы для миграции

См. Руководство по миграции для получения дополнительной информации.

tf.compat.v1.io.write_graph, tf.compat.v1.train.write_graph, `tf.compat.v2.io.write_graph`

tf.io.write_graph(
    graph_or_graph_def, logdir, name, as_text=True
)

Граф записывается как текстовый протокол, если as_text False.

v = tf.Variable(0, name='my_variable')
sess = tf.compat.v1.Session()
tf.io.write_graph(sess.graph_def, '/tmp/my-model', 'train.pbtxt')

или

v = tf.Variable(0, name='my_variable')
sess = tf.compat.v1.Session()
tf.io.write_graph(sess.graph, '/tmp/my-model', 'train.pbtxt')
Аргументы
graph_or_graph_def Graph или GraphDef протокол буфера.
logdir Каталог, куда необходимо записать граф. Он может ссылаться на удаленные файловые системы, такие как Google Cloud Storage (GCS).
name Имя файла для графа.
as_text Если True, записывает граф как ASCII протокол.
Возвращаемое значение
Путь к выходному файлу протокола.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/io/write_graph

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API