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新手实现2048小游戏
Mar 31 Python
在Python中定义和使用抽象类的方法
Jun 30 Python
python xml解析实例详解
Nov 14 Python
python+mongodb数据抓取详细介绍
Oct 25 Python
Python基于OpenCV实现视频的人脸检测
Jan 23 Python
详解python实现线程安全的单例模式
Mar 05 Python
Python扩展内置类型详解
Mar 26 Python
Python一个简单的通信程序(客户端 服务器)
Mar 06 Python
python中selenium操作下拉滚动条的几种方法汇总
Jul 14 Python
numpy.array 操作使用简单总结
Nov 08 Python
Python如何使用BeautifulSoup爬取网页信息
Nov 26 Python
python 实现快速生成连续、随机字母列表
Nov 28 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
php在程序中将网页生成word文档并提供下载的代码
2012/10/09 PHP
微信支付PHP SDK之微信公众号支付代码详解
2015/12/09 PHP
PHP编程求最大公约数与最小公倍数的方法示例
2017/05/29 PHP
JQuery事件e参数的方法preventDefault()取消默认行为
2013/09/26 Javascript
用js正确判断用户名cookie是否存在的方法
2014/01/28 Javascript
JavaScript设计模式之工厂方法模式介绍
2014/12/28 Javascript
js获取域名的方法
2015/01/27 Javascript
CSS或者JS实现鼠标悬停显示另一元素
2016/01/22 Javascript
jQuery自定义图片上传插件实例代码
2017/04/04 jQuery
js学习心得_一个简单的动画库封装tween.js
2017/07/14 Javascript
基于zTree树形菜单的使用实例
2017/12/25 Javascript
基于vue 添加axios组件,解决post传参数为null的问题
2018/03/05 Javascript
vue移动UI框架滑动加载数据的方法
2018/03/12 Javascript
NodeJS父进程与子进程资源共享原理与实现方法
2018/03/16 NodeJs
微信小程序支付功能 php后台对接完整代码分享
2018/06/12 Javascript
Vue在页面数据渲染完成之后的调用方法
2018/09/11 Javascript
JavaScript实现的鼠标跟随特效示例【2则实例】
2018/12/22 Javascript
记一次vue-webpack项目优化实践详解
2019/02/17 Javascript
微信小程序之 catalog 切换实现解析
2019/09/12 Javascript
解决vue使用vant下拉框van-dropdown-item 绑定title值不变问题
2020/08/05 Javascript
解决在Vue中使用axios POST请求变成OPTIONS的问题
2020/08/14 Javascript
用Python写飞机大战游戏之pygame入门(4):获取鼠标的位置及运动
2015/11/05 Python
在pycharm中debug 实时查看数据操作(交互式)
2020/06/09 Python
Python fileinput模块如何逐行读取多个文件
2020/10/05 Python
使用Pytorch搭建模型的步骤
2020/11/16 Python
python代码实现猜拳小游戏
2020/11/30 Python
资生堂美国官网:Shiseido美国
2016/09/02 全球购物
自我鉴定 电子商务专业
2014/01/30 职场文书
银行贷款承诺书
2014/03/29 职场文书
高校师德师风自我剖析材料
2014/09/29 职场文书
婚宴邀请函
2015/01/30 职场文书
2015年世界环境日活动总结
2015/02/11 职场文书
傲慢与偏见电影观后感
2015/06/10 职场文书
新闻稿件写作技巧
2015/07/18 职场文书
HAM-2000摩机图
2021/04/22 无线电
threejs太阳光与阴影效果实例代码
2022/04/05 Javascript