TensorFlow中tf.batch_matmul()的用法

更新时间：2021年06月02日 08:37:16 作者：yyhhlancelot

这篇文章主要介绍了TensorFlow中tf.batch_matmul()的用法，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教

TensorFlow中tf.batch_matmul()用法

如果有两个三阶张量，size分别为

a.shape = [100, 3, 4]
b.shape = [100, 4, 5]
c = tf.batch_matmul(a, b)

则c.shape = [100, 3, 5] //将每一对 3x4 的矩阵与 4x5 的矩阵分别相乘。batch_size不变

100为张量的batch_size。剩下的两个维度为数据的维度。

不过新版的tensorflow已经移除了上面的函数，使用时换为tf.matmul就可以了。与上面注释的方式是同样的。

附：如果是更高维度。例如(a, b, m, n) 与(a, b, n, k)之间做matmul运算。则结果的维度为(a, b, m, k)。

TensorFlow如何实现batch_matmul

我们知道，在tensorflow早期版本中有tf.batch_matmul()函数，可以实现多维tensor和低维tensor的直接相乘，这在使用过程中非常便捷。

但是最新版本的tensorflow现在只有tf.matmul()函数可以使用，不过只能实现同维度的tensor相乘, 下面的几种方法可以实现batch matmul的可能。

例如: tensor A(batch_size,m,n), tensor B(n,k),实现batch matmul 使得A * B。

方法1: 利用tf.matmul()

对tensor B 进行增维和扩展

A = tf.Variable(tf.random_normal(shape=(batch_size, 2, 3)))
B = tf.Variable(tf.random_normal(shape=(3, 5)))
B_exp = tf.tile(tf.expand_dims(B,0),[batch_size, 1, 1]) #先进行增维再扩展
C = tf.matmul(A, B_exp)

方法2: 利用tf.reshape()

对tensor A 进行reshape操作，然后利用tf.matmul()

A = tf.Variable(tf.random_normal(shape=(batch_size, 2, 3)))
B = tf.Variable(tf.random_normal(shape=(3, 5)))
A = tf.reshape(A, [-1, 3])
C = tf.reshape(tf.matmul(A, B), [-1, 2, 5])

方法3: 利用tf.scan()

利用tf.scan() 对tensor按第0维进行展开的特性

A = tf.Variable(tf.random_normal(shape=(batch_size, 2, 3)))
B = tf.Variable(tf.random_normal(shape=(3, 5)))
initializer = tf.Variable(tf.random_normal(shape=(2,5)))
C = tf.scan(lambda a,x: tf.matmul(x, B), A, initializer)

方法4: 利用tf.einsum()

A = tf.Variable(tf.random_normal(shape=(batch_size, 2, 3)))
B = tf.Variable(tf.random_normal(shape=(3, 5)))
C = tf.einsum('ijk,kl->ijl',A,B)

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

python模拟键盘输入切换键盘布局过程解析
这篇文章主要介绍了python模拟键盘输入切换键盘布局过程解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2019-08-08
对Python生成汉字字库文字,以及转换为文字图片的实例详解
今天小编就为大家分享一篇对Python生成汉字字库文字,以及转换为文字图片的实例详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-01-01
Python实现12种降维算法的示例代码
数据降维算法是机器学习算法中的大家族，与分类、回归、聚类等算法不同，它的目标是将向量投影到低维空间，以达到某种目的如可视化，或是做分类。本文将利用Python实现12种降维算法，需要的可以参考一下
2022-04-04
Python的列表和元组详情
这篇文章主要介绍了Python的列表和元组，列表和元组是python组常见的内置内省，下面文章我们讲围绕Python的列表和元组的相关资料展开话题,感兴趣的小伙伴以参考一下
2021-10-10
Python可视化神器pyecharts之绘制箱形图
这篇文章主要介绍了Python可视化神器pyecharts之绘制箱形图，箱形图（Box-plot）又称为盒须图、盒式图或箱线图，是一种用作显示一组数据分散情况资料的统计图，因形状如箱子而得名
2022-07-07
关于Python中浮点数精度处理的技巧总结
双精度浮点数(double)是计算机使用的一种数据类型，使用 64 位（8字节）来存储一个浮点数。下面这篇文章主要给大家总结介绍了关于Python中浮点数精度处理的技巧，需要的朋友可以参考借鉴，下面来一起看看吧。
2017-08-08
python 生成空字符串的5种方法
有时候我们需要生成一个空的字符串,本文就来介绍一下python 生成空字符串的5种方法,包括使用空的单引号或双引号、使用str函数、字符串连接、字符串格式化以及字符串乘法,感兴趣的可以了解一下
2024-01-01
Python 提取dict转换为xml/json/table并输出的实现代码
这篇文章主要介绍了Python 提取dict转换为xml/json/table并输出的实现代码,需要的朋友可以参考下
2016-08-08
python获取本机外网ip的方法
这篇文章主要介绍了python获取本机外网ip的方法,可实现从外网显示IP的网站获取本机IP的功能,非常具有实用价值,需要的朋友可以参考下
2015-04-04
Django Sitemap 站点地图的实现方法
这篇文章主要介绍了Django Sitemap 站点地图的实现方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2019-04-04