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.]]
短小精悍算例:TensorFlow通过定义变量实现矩阵乘法
猜你喜欢
转载自blog.csdn.net/weixin_39464400/article/details/106109162
今日推荐
周排行