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


Posted in Python onJuly 27, 2018

如下所示:

# 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 相关文章推荐
wxPython 入门教程
Oct 07 Python
用Python的Tornado框架结合memcached页面改善博客性能
Apr 24 Python
python pandas中DataFrame类型数据操作函数的方法
Apr 08 Python
python将字符串以utf-8格式保存在txt文件中的方法
Oct 30 Python
详解如何用django实现redirect的几种方法总结
Nov 22 Python
基于Django统计博客文章阅读量
Oct 29 Python
Python实现ATM系统
Feb 17 Python
python 装饰器功能与用法案例详解
Mar 06 Python
利用Python将图片中扭曲矩形的复原
Sep 07 Python
Python3+Appium安装及Appium模拟微信登录方法详解
Feb 16 Python
python中Matplotlib绘制直线的实例代码
Jul 04 Python
Python的property属性详细讲解
Apr 11 Python
对Tensorflow中的矩阵运算函数详解
Jul 27 #Python
python+influxdb+shell编写区域网络状况表
Jul 27 #Python
TensorFlow 合并/连接数组的方法
Jul 27 #Python
pytorch训练imagenet分类的方法
Jul 27 #Python
使用pytorch进行图像的顺序读取方法
Jul 27 #Python
PyTorch读取Cifar数据集并显示图片的实例讲解
Jul 27 #Python
pytorch 数据集图片显示方法
Jul 26 #Python
You might like
自动把纯文本转换成Web页面的php代码
2009/08/27 PHP
PHP数据过滤的方法
2013/10/30 PHP
php类声明和php类使用方法示例分享
2014/03/29 PHP
php封装好的人民币数值转中文大写类
2015/12/20 PHP
php each 返回数组中当前的键值对并将数组指针向前移动一步实例
2016/11/22 PHP
tp5.1 框架数据库高级查询技巧实例总结
2020/05/25 PHP
AngularJS 整理一些优化的小技巧
2016/08/18 Javascript
javascript轮播图算法
2016/10/21 Javascript
原生js实现可拖拽效果
2017/02/28 Javascript
jquery.validate表单验证插件使用详解
2017/06/21 jQuery
angular第三方包开发整理(小结)
2018/04/19 Javascript
Vue.js点击切换按钮改变内容的实例讲解
2018/08/22 Javascript
vue 音乐App QQ音乐搜索列表最新接口跨域设置方法
2018/09/25 Javascript
vue tab滚动到一定高度,固定在顶部,点击tab切换不同的内容操作
2020/07/22 Javascript
[01:01:51]EG vs VG Supermajor小组赛B组 BO3 第二场 6.2
2018/06/03 DOTA
python实现基于两张图片生成圆角图标效果的方法
2015/03/26 Python
详解Python中内置的NotImplemented类型的用法
2015/03/31 Python
Python用模块pytz来转换时区
2016/08/19 Python
python中logging包的使用总结
2018/02/28 Python
python保存数据到本地文件的方法
2018/06/23 Python
局域网内python socket实现windows与linux间的消息传送
2019/04/19 Python
浅谈python 中类属性共享的问题
2019/07/02 Python
对Python中小整数对象池和大整数对象池的使用详解
2019/07/09 Python
Python 实现遥感影像波段组合的示例代码
2019/08/04 Python
如何利用python给图片添加半透明水印
2019/09/06 Python
python dataframe NaN处理方式
2019/12/26 Python
浅谈Python3实现两个矩形的交并比(IoU)
2020/01/18 Python
python 用opencv实现霍夫线变换
2020/11/27 Python
详解HTML5中的标签
2015/06/19 HTML / CSS
将HTML5 Canvas的内容保存为图片借助toDataURL实现
2013/05/20 HTML / CSS
基于MUI框架使用HTML5实现的二维码扫描功能
2018/03/01 HTML / CSS
西班牙汉普顿小姐:购买帆布鞋和太阳镜
2016/10/23 全球购物
大学生的网上创业计划书
2013/12/31 职场文书
优秀德育工作者事迹材料
2014/05/07 职场文书
公务员个人年终总结
2015/02/12 职场文书
公司慰问信范文
2015/03/23 职场文书