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 相关文章推荐
Python实现3行代码解简单的一元一次方程
Aug 18 Python
Python EOL while scanning string literal问题解决方法
Sep 18 Python
python计算对角线有理函数插值的方法
May 07 Python
Python实现批量下载图片的方法
Jul 08 Python
linux环境下python中MySQLdb模块的安装方法
Jun 16 Python
python日期时间转为字符串或者格式化输出的实例
May 29 Python
Caffe均值文件mean.binaryproto转mean.npy的方法
Jul 09 Python
Python使用folium excel绘制point
Jan 03 Python
WIn10+Anaconda环境下安装PyTorch(避坑指南)
Jan 30 Python
python剪切视频与合并视频的实现
Mar 03 Python
在 Windows 下搭建高效的 django 开发环境的详细教程
Jul 27 Python
MoviePy常用剪辑类及Python视频剪辑自动化
Dec 18 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
ThinkPHP3.1的Widget新用法
2014/06/19 PHP
Zend Framework实现具有基本功能的留言本(附demo源码下载)
2016/03/22 PHP
php redis实现文章发布系统(用户投票系统)
2017/03/04 PHP
js停止输出代码
2008/07/20 Javascript
Jquery 弹出层插件实现代码
2009/10/24 Javascript
jQuery中使用了document和window哪些属性和方法小结
2011/09/13 Javascript
JS函数实现动态添加CSS样式表文件
2012/12/15 Javascript
JS 跳转页面延迟2种方法
2013/03/29 Javascript
jquery图片不完全按比例自动缩小的简单代码
2013/07/29 Javascript
jQuery实现返回顶部功能适合不支持js的浏览器
2014/08/19 Javascript
精通JavaScript的this关键字
2020/05/28 Javascript
判断是否存在子节点的实现代码
2016/05/18 Javascript
JavaScript组成、引入、输出、运算符基础知识讲解
2016/12/08 Javascript
使用nodejs下载风景壁纸
2017/02/05 NodeJs
详解angularjs获取元素以及angular.element()用法
2017/07/25 Javascript
微信小程序之swiper轮播图中的图片自适应高度的方法
2018/04/23 Javascript
关于单文件组件.vue的使用
2018/09/20 Javascript
浅析vue-router原理
2018/10/19 Javascript
学习使用ExpressJS 4.0中的新Router的用法
2018/11/06 Javascript
vue路由前进后退动画效果的实现代码
2018/12/10 Javascript
浅谈Angular单元测试总结
2019/03/22 Javascript
JavaScript基于面向对象实现的无缝滚动轮播示例
2020/01/17 Javascript
vue cli3适配所有端方案的实现
2020/04/13 Javascript
Python中变量交换的例子
2014/08/25 Python
Python 列表排序方法reverse、sort、sorted详解
2016/01/22 Python
win10环境下python3.5安装步骤图文教程
2017/02/03 Python
Python实现分段线性插值
2018/12/17 Python
python调用虹软2.0第三版的具体使用
2019/02/22 Python
由面试题加深对Django的认识理解
2019/07/19 Python
python实现飞机大战小游戏
2019/11/08 Python
python 实现两个线程交替执行
2020/05/02 Python
如何使用PyCharm引入需要使用的包的方法
2020/09/22 Python
html5教程画矩形代码分享
2013/12/04 HTML / CSS
房屋继承公证书
2014/04/10 职场文书
出差报告怎么写
2014/11/06 职场文书
党支部鉴定意见
2015/06/02 职场文书