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

2025-03-15 08:44:07

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。
猜你喜欢