Tensorflow架构中的matmul矩阵乘法和mul点乘法

一般来说会遇到这两种乘法,这两种乘法是我们经常会遇到的,不过在计算中,矩阵乘法会更常用一点

例如,我设置两个常量,分别是v1和v2,现在需要将两者相乘

v1 = tf.constant([[1.0,2.0],[3.0,4.0]])
v2 = tf.constant([[5.0,6.0],[7.0,8.0]])

点乘和矩阵乘法两者计算区别如下:

1.采用点乘,也就是点对点乘法,得到的结果是

[[ 5. 12.]
 [21. 32.]]

 

2.采用矩阵乘法

[[19. 22.]
 [43. 50.]]

 

算法上来说是这样的

 

生成的两个结果都是float32的类型

 

点乘和矩阵乘法一样,如果两个矩阵不满足(A1,B1) (B1,C1)条件的话,就会计算出错

 

完整程序:

#演示矩阵乘法
import tensorflow as tf
batch_size = 8

v1 = tf.constant([[1.0,2.0],[3.0,4.0]])
v2 = tf.constant([[5.0,6.0],[7.0,8.0]])

a = tf.matmul(v1,v2)
b = v1 * v2

sess = tf.Session()
init_op = tf.global_variables_initializer()
sess.run(init_op)

print(sess.run(a))
print(sess.run(b))

 

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值