短小精悍算例：TensorFlow实现两个矩阵的相乘运算

最新推荐文章于 2021-08-18 22:31:08 发布

DreamByter

最新推荐文章于 2021-08-18 22:31:08 发布

阅读量800

点赞数

分类专栏： Python 机器学习文章标签： tensorflow python numpy

本文链接：https://blog.csdn.net/weixin_39464400/article/details/106108807

版权

Python 同时被 2 个专栏收录

23 篇文章 0 订阅

订阅专栏

机器学习

23 篇文章 0 订阅

订阅专栏

import tensorflow as tf
import numpy as np
a1 = tf.constant(np.ones([4, 4]))
a2 = tf.constant(np.ones([4, 4])*0.3)
sess = tf.Session()      # 开启一个会话
print(sess.run(a1),'\n') # 打印矩阵a1
print(sess.run(a2),'\n') # 打印矩阵a2
print(sess.run(a1@a2))   # 打印矩阵a1*a2

输出结果：
[[1. 1. 1. 1.]
 [1. 1. 1. 1.]
 [1. 1. 1. 1.]
 [1. 1. 1. 1.]] 

[[0.3 0.3 0.3 0.3]
 [0.3 0.3 0.3 0.3]
 [0.3 0.3 0.3 0.3]
 [0.3 0.3 0.3 0.3]] 

[[1.2 1.2 1.2 1.2]
 [1.2 1.2 1.2 1.2]
 [1.2 1.2 1.2 1.2]
 [1.2 1.2 1.2 1.2]]