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]]
短小精悍算例:TensorFlow实现两个矩阵的相乘运算
猜你喜欢
转载自blog.csdn.net/weixin_39464400/article/details/106108807
今日推荐
周排行