TensorFlow 定义m行n列 矩阵

定义a为1行5列的矩阵，b为5行一列的矩阵：

a = tf.constant([[1,2,3,4,5]])
b = tf.constant([[1],
[2],
[3],
[4],
[5]])

满足矩阵相乘的条件，a*b为一个1行1列的矩阵，b*a为5行5列的矩阵：

with tf.Session() as sess:
　　print(sess.run(tf.matmul(b, a)))

　　print("")
　　print(sess.run(tf.matmul(a, b)))

 

输出：

[[ 1 2 3 4 5]
[ 2 4 6 8 10]
[ 3 6 9 12 15]
[ 4 8 12 16 20]
[ 5 10 15 20 25]]

[[55]]

 

转载于:https://www.cnblogs.com/wjbyzsa/p/9612620.html