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下读取公私钥做加解密实例详解
Mar 29 Python
python实现树形打印目录结构
Mar 29 Python
Django中使用Whoosh进行全文检索的方法
Mar 31 Python
Django框架使用内置方法实现登录功能详解
Jun 12 Python
利用python、tensorflow、opencv、pyqt5实现人脸实时签到系统
Sep 25 Python
vim自动补全插件YouCompleteMe(YCM)安装过程解析
Oct 21 Python
python实现小世界网络生成
Nov 21 Python
Python实现平行坐标图的绘制(plotly)方式
Nov 22 Python
OpenCV Python实现拼图小游戏
Mar 23 Python
使用keras根据层名称来初始化网络
May 21 Python
Python collections.defaultdict模块用法详解
Jun 18 Python
Python用dilb提取照片上人脸的示例
Oct 26 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将对象转换成数组的方法(兼容多维数组类型)
2013/06/21 PHP
PHP循环函数使用介绍之PHP基础入门教程
2013/09/21 PHP
php命令行(cli)模式下报require 加载路径错误的解决方法
2015/11/23 PHP
php处理json格式数据经典案例总结
2016/05/19 PHP
基于Laravel实现的用户动态模块开发
2017/09/21 PHP
javascript入门·对象属性方法大总结
2007/10/01 Javascript
js动态控制table的tr、td增加及删除的具体实现
2014/04/30 Javascript
js简单实现点击左右运动的方法
2015/04/10 Javascript
js日期范围初始化得到前一个月日期的方法
2015/05/05 Javascript
常用jQuery代码分享
2015/07/14 Javascript
JS实现的不规则TAB选项卡效果代码
2015/09/18 Javascript
JavaScript模版引擎的基本实现方法浅析
2016/02/15 Javascript
基于代数方程库Algebra.js解二元一次方程功能示例
2017/06/09 Javascript
浅谈手写node可读流之流动模式
2018/06/01 Javascript
从零搭一个自用的前端脚手架的方法步骤
2019/09/23 Javascript
js实现随机抽奖
2020/03/19 Javascript
vue+AI智能机器人回复功能实现
2020/07/16 Javascript
[42:20]Winstrike vs VGJ.S 2018国际邀请赛淘汰赛BO3 第二场 8.23
2018/08/24 DOTA
python发送邮件示例(支持中文邮件标题)
2014/02/16 Python
Python3实现将文件归档到zip文件及从zip文件中读取数据的方法
2015/05/22 Python
python常见排序算法基础教程
2017/04/13 Python
python中学习K-Means和图片压缩
2017/11/20 Python
对Python闭包与延迟绑定的方法详解
2019/01/07 Python
萨克斯第五大道的折扣店:Saks Fifth Avenue OFF 5TH
2016/08/25 全球购物
里程积分管理买卖交换平台:Points.com
2017/01/13 全球购物
精选鞋类、服装和配饰的全球领先目的地:Bodega
2021/02/27 全球购物
2013年入党人员的自我鉴定
2013/10/25 职场文书
党委书记岗位职责
2013/11/24 职场文书
建筑系毕业生自我鉴定
2014/01/24 职场文书
《最大的“书”》教学反思
2014/02/14 职场文书
化学教学随笔感言
2014/02/19 职场文书
读书活动总结范文
2014/04/26 职场文书
服务口号大全
2014/06/11 职场文书
党员转正申请报告
2015/05/15 职场文书
辞职离别感言
2015/08/04 职场文书
毕业欢送晚会主持词
2019/06/25 职场文书