example.py

import tensorflow as tf
a = tf.Graph()
input1 = tf.placeholder(tf.float32, shape=(2))
input2 = tf.placeholder(tf.float32, shape=(2))
output = tf.mul(input1, input2)

with tf.Session() as sess:
  print(sess.run([output], feed_dict={input1:[7,2], input2:[2,4]}))
  tf.train.write_graph(sess.graph_def, 'models/', 'test_graph_multi_dim.pb', as_text=True)