1. 程式人生 > >TensorFlow 定義m行n列 矩陣

TensorFlow 定義m行n列 矩陣

[1] orf tensor ESS a* atm sta 矩陣 一個

定義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]]

TensorFlow 定義m行n列 矩陣