Function main

tensorflow/contrib/graph_editor/examples/edit_graph_example.py:30–50

(_)

def main(_):
  # create a graph
  g = tf.Graph()
  with g.as_default():
    a = tf.constant(1.0, shape=[2, 3], name="a")
    b = tf.constant(2.0, shape=[2, 3], name="b")
    c = tf.add(
        tf.placeholder(dtype=np.float32),
        tf.placeholder(dtype=np.float32),
        name="c")

  # modify the graph
  ge.swap_inputs(c.op, [a, b])

  # print the graph def
  print(g.as_graph_def())

  # and print the value of c
  with tf.Session(graph=g) as sess:
    res = sess.run(c)
    print(res)


if __name__ == "__main__":