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

2025-09-25 07:12:53

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)

声明:本网站引用、摘录或转载内容仅供网站访问者交流或参考,不代表本站立场,如存在版权或非法内容,请联系站长删除,联系邮箱:site.kefu@qq.com。
猜你喜欢