/
NikolayIvkin
/
tutorials1
Обзор
Документация
Войти
/
NikolayIvkin
/
tutorials1
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
tensorflow-java/src/main/python/tensorflowGraph.py
16 строк
572 B
Eric Martin
Merge pull request #8125 from eugenp/revert-8119-BAEL-3275-2
01 ноя 2019, 04:43
01 ноя 2019, 04:43
3225470
Код
Авторство
О чём код?
import tensorflow as tf graph = tf.Graph() builder = tf.saved_model.builder.SavedModelBuilder('./model') writer = tf.summary.FileWriter('.') with graph.as_default(): a = tf.constant(2, name='a') b = tf.constant(3, name='b') x = tf.placeholder(tf.int32, name='x') y = tf.placeholder(tf.int32, name='y') z = tf.math.add(a*x, b*y, name='z') writer.add_graph(tf.get_default_graph()) writer.flush() sess = tf.Session() sess.run(z, feed_dict = {x: 2, y: 3}) builder.add_meta_graph_and_variables(sess, [tf.saved_model.tag_constants.SERVING]) builder.save()