tensorflow 用矩阵运算替换for循环用tf.tile而不写for的方法

更新时间：2018年07月27日 09:56:39 作者：guotong1988

今天小编就为大家分享一篇tensorflow 用矩阵运算替换for循环用tf.tile而不写for的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

如下所示：

# u [32,30,200]
# u_logits [400,32,30]
q_j_400 = [] 
for j in range(400):
 q_j_400.append(tf.squeeze(tf.matmul(tf.transpose(u，[0,2,1]),tf.expand_dims(tf.nn.softmax(u_logits[j]),-1)),[2])) # tf.matmul [32,200,30],[32,30,1]
test_result = tf.stack(q_j_400)
test_result = tf.transpose(test_result,[1,0,2])

可以通过tf.tile实现更高速的版本

# u [32,30,200]
# u_logits [32,400,30]
u_tile = tf.tile(tf.expand_dims(u,1),[1,400,1,1])
u_logits = tf.expand_dims(tf.nn.softmax(u_logits,-1),-1)
test_result = tf.reduce_sum(u_logits * u_tile,-2) # [32,400,30,1]*[32,400,30,200]

以上这篇tensorflow 用矩阵运算替换for循环用tf.tile而不写for的方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

Python中replace方法实例分析
这篇文章主要介绍了Python中replace方法,有一定的实用价值,需要的朋友可以参考下
2014-08-08
Python如何运用pyaudio库去做一个固定采样率音频录制器
这篇文章主要介绍了Python如何运用pyaudio库去做一个固定采样率音频录制器问题，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2023-05-05
教你用Python写一个植物大战僵尸小游戏
这篇文章主要介绍了教你用Python写一个植物大战僵尸小游戏,文中有非常详细的代码示例,对正在学习python的小伙伴们有非常好的帮助,需要的朋友可以参考下
2021-04-04
PyCharm2019安装教程及其使用(图文教程)
这篇文章主要介绍了PyCharm2019安装教程(图文教程)，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2019-09-09
使用Python读写文本文件及编写简单的文本编辑器
这篇文章主要介绍了使用Python读写文本文件及编写简单的文本编辑器的方法,可以利用简单的Python代码实现编辑器中的读取清空和文本文件的新建等操作,需要的朋友可以参考下
2016-03-03
GraphQL在Django中的使用教程
这篇文章主要介绍了GraphQL在Django中的使用教程,本文结合示例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2022-12-12
python 定时器,实现每天凌晨3点执行的方法
今天小编就为大家分享一篇python 定时器,实现每天凌晨3点执行的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-02-02
查看python安装路径及pip安装的包列表及路径
这篇文章主要介绍了查看python安装路径及pip安装的包列表及路径,本文给大家介绍的非常详细，具有一定的参考借鉴价值，需要的朋友可以参考下
2019-04-04
python skimage图像处理
这篇文章主要为大家介绍了python skimage图像处理，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2022-06-06
详解如何管理多个Python版本和虚拟环境
这篇文章主要介绍了详解如何管理多个Python版本和虚拟环境，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2019-05-05