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的方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

pytorch预测之解决多次预测结果不一致问题
这篇文章主要介绍了pytorch多次预测结果不一致的解决方案，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2021-06-06
Python运行的17个时新手常见错误小结
当初学 Python 时，想要弄懂 Python 的错误信息的含义可能有点复杂。这里列出了常见的的一些让你程序 crash 的运行时错误
2012-08-08
python logging模块的使用
这篇文章主要介绍了python logging模块的使用，帮助大家更好的理解和学习python，感兴趣的朋友可以了解下
2020-09-09
python解压zip包中文乱码解决方法
这篇文章主要介绍了python解压zip包中文乱码解决方法，帮助大家更好的理解和学习python，感兴趣的朋友可以了解下
2020-11-11
Python如何实现机器人聊天
这篇文章主要介绍了Python如何实现机器人聊天，帮助大家更好的理解和学习python，感兴趣的朋友可以了解下
2020-09-09
使用 NumPy 和 Matplotlib 绘制函数图
Matplotlib 是 Python 的绘图库。它可与 NumPy 一起使用，提供了一种有效的 MatLab 开源替代方案。它也可以和图形工具包一起使用，如 PyQt 和 wxPython
2021-09-09
Python机器学习入门(五)之Python算法审查
这篇文章主要介绍了Python机器学习入门知识，本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2021-08-08
Python Asyncio调度原理详情
这篇文章主要介绍了Python Asyncio调度原理详情，Python.Asyncio是一个大而全的库，它包括很多功能，而跟核心调度相关的逻辑除了三种可等待对象外，还有其它一些功能，它们分别位于runners.py，base_event.py，event.py三个文件中
2022-06-06
EM算法的python实现的方法步骤
本篇文章主要介绍了EM算法的python实现的方法步骤，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2018-01-01
Python 内置方法和属性详解
这篇文章主要为大家介绍了Python 内置方法和属性，具有一定的参考价值，感兴趣的小伙伴们可以参考一下，希望能够给你带来帮助
2021-12-12