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

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]]

猜你喜欢

转载自blog.csdn.net/weixin_39464400/article/details/106108807