短小精悍算例:TensorFlow通过定义变量实现矩阵乘法

import tensorflow as tf
import numpy as np
#定义变量
b1 = tf.Variable(np.ones([4, 4])*2)
b2 = tf.Variable(np.ones([4, 4]))
#定义矩阵乘法
b1_elementwise_b2 = b1 * b2    # 对应元素相乘
b1_dot_b2 = tf.matmul(b1, b2)  # 矩阵乘法
#variable需要初始化
init = tf.global_variables_initializer()
sess = tf.Session()
sess.run(init)
#打印结果
print(sess.run(b1),'\n')
print(sess.run(b2),'\n')
print(sess.run(b1_elementwise_b2),'\n')
print(sess.run(b1_dot_b2))

输出结果:
[[2. 2. 2. 2.]
 [2. 2. 2. 2.]
 [2. 2. 2. 2.]
 [2. 2. 2. 2.]] 

[[1. 1. 1. 1.]
 [1. 1. 1. 1.]
 [1. 1. 1. 1.]
 [1. 1. 1. 1.]] 

[[2. 2. 2. 2.]
 [2. 2. 2. 2.]
 [2. 2. 2. 2.]
 [2. 2. 2. 2.]] 

[[8. 8. 8. 8.]
 [8. 8. 8. 8.]
 [8. 8. 8. 8.]
 [8. 8. 8. 8.]]

猜你喜欢

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