基于tensorflow的矩阵的乘法(python3.6)

时间:2026-02-14 12:23:30

1、加载tensorflow模块:

import tensorflow as tf

基于tensorflow的矩阵的乘法(python3.6)

2、创建一个1*2的矩阵:

mat1 = tf.constant([[3., 3.]])

基于tensorflow的矩阵的乘法(python3.6)

3、创建一个2*1的矩阵:

mat2 = tf.constant([[2.],[2.6]])

基于tensorflow的矩阵的乘法(python3.6)

4、计算矩阵的乘积:

p = tf.matmul(mat1, mat2)

得到一个1*1的矩阵。

基于tensorflow的矩阵的乘法(python3.6)

5、如果想查看p的具体数值,需要建立一个会话:

sess = tf.Session()

result = sess.run(p)

print(result)

关闭会话:

sess.close()

基于tensorflow的矩阵的乘法(python3.6)

6、如果:

p = tf.matmul(mat2, mat1)

就得到一个2*2的矩阵。

基于tensorflow的矩阵的乘法(python3.6)

© 2026 海能知识库
信息来自网络 所有数据仅供参考
有疑问请联系站长 site.kefu@gmail.com