用tensorflow计算矩阵中行向量间的欧式距离

import tensorflow as tf
a=tf.constant([[1,1.,3],[2,3,4]])
b=tf.constant([[1,1,1.],[2,2,2]])
c=tf.sqrt(tf.reduce_sum(tf.square(a-b),axis=1))
with tf.Session() as sess:
    print(sess.run(a))
    print(sess.run(b))
    print(sess.run(c))