1. 程式人生 > >tf.matmul()研究【Python】

tf.matmul()研究【Python】

含義

  • 表示矩陣乘法

經典例項

import tensorflow as tf
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
# 建立一個常量 op , 產生一個 1*2 的矩陣.
matrix1 = tf.constant([[3., 3.]])
# 建立另外一個常量 op, 產生一個 2x1 矩陣.
matrix2 = tf.constant([[2.], [2.]])
product = tf.matmul(matrix1, matrix2)

# 建立一個session,並提交執行
sess = tf.Session()
result =
sess.run(product) print(result) sess.close()

輸出的結果是:

[[12.]]