In [2]:
import numpy as np
import tensorflow as tf
import tensorboard_jupyter as tb
sess = tf.Session()
my_array = np.array([[1., 3., 5., 7., 9.],
[2., 0., 2., 4., 6.],
[-6., -3., 0., 3., 6.]])
# 入力2つ分にするため複製
x_vals = np.array([my_array, my_array + 1])
x_data = tf.placeholder(tf.float32, shape=(3,5))
# 乗算、加算用
m1 = tf.constant([[1.],[0.],[-1.],[2.],[4.]])
m2 = tf.constant([[2.]])
a1 = tf.constant([[10.]])
prod1 = tf.matmul(x_data, m1)
prod2 = tf.matmul(prod1, m2)
add1 = tf.add(prod2, a1)
for x_val in x_vals:
print(sess.run(add1, feed_dict={x_data: x_val}))
tf.summary.FileWriter('./log/', sess.graph)
tb.show_graph(tf.get_default_graph().as_graph_def())
In [ ]: