Tensorflowのsess.run()で複数の値を取得する方法

以下のサイトを参考にsess.run()で複数の値を取得する方法を確認した。

このサイトにはTensorflowの計算グラフのことについても説明されているので、非常に参考になる。

sess.run()で複数の値を取得するには以下の様に記述する。

const1 = tf.constant(2)

const2 = tf.constant(3)

add_op = tf.add(const1, const2)

mul_op = tf.mul(add_op, const2)

with tf.Session() as sess:

  result, result2 = sess.run([mul_op, add_op])

  print(result)

  print(result2)

このようにrunの引数にオペレーションのリストを渡せば、複数のオペレーションを実行することもできる。