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连接mysql实例分享
Oct 09 Python
requests和lxml实现爬虫的方法
Jun 11 Python
使用python 和 lint 删除项目无用资源的方法
Dec 20 Python
Python网络爬虫之爬取微博热搜
Apr 18 Python
在Python中COM口的调用方法
Jul 03 Python
Django学习之文件上传与下载
Oct 06 Python
如何通过python实现人脸识别验证
Jan 17 Python
python实现打砖块游戏
Feb 25 Python
python GUI库图形界面开发之PyQt5滚动条控件QScrollBar详细使用方法与实例
Mar 06 Python
python实现UDP协议下的文件传输
Mar 20 Python
python由已知数组快速生成新数组的方法
Apr 08 Python
Python-for循环的内部机制
Jun 12 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
php2html php生成静态页函数
2008/12/08 PHP
ThinkPHP视图查询详解
2014/06/30 PHP
Yii中的relations数据关联查询及统计功能用法详解
2016/07/14 PHP
Yii2中事务的使用实例代码详解
2016/09/07 PHP
二级域名或跨域共享Cookies的实现方法
2008/08/07 Javascript
css+js实现部分区域高亮可编辑遮罩层
2014/03/04 Javascript
AngularJS表单编辑提交功能实例
2015/02/13 Javascript
jQuery简单实现两级下拉菜单效果代码
2015/09/15 Javascript
Node.js检测端口(port)是否被占用的简单示例
2016/09/29 Javascript
js实现文字选中分享功能
2017/01/25 Javascript
JS前端开发判断是否是手机端并跳转操作(小结)
2017/02/05 Javascript
Angularjs中的ui-bootstrap的使用教程
2017/02/19 Javascript
vue登录注册及token验证实现代码
2017/12/14 Javascript
javascript数据结构之多叉树经典操作示例【创建、添加、遍历、移除等】
2018/08/01 Javascript
js html实现计算器功能
2018/11/13 Javascript
koa2 从入门到精通(小结)
2019/07/23 Javascript
webpack+vue-cil 中proxyTable配置接口地址代理操作
2020/07/18 Javascript
使用Python脚本来控制Windows Azure的简单教程
2015/04/16 Python
numpy中索引和切片详解
2017/12/15 Python
基于python绘制科赫雪花
2018/06/22 Python
python实现爬山算法的思路详解
2019/04/09 Python
python做反被爬保护的方法
2019/07/01 Python
python selenium登录豆瓣网过程解析
2019/08/10 Python
python多进程 主进程和子进程间共享和不共享全局变量实例
2020/04/25 Python
python中加背景音乐如何操作
2020/07/19 Python
Python使用xpath实现图片爬取
2020/09/16 Python
HTML5新增元素如何兼容旧浏览器有哪些方法
2014/05/09 HTML / CSS
Html5新增标签有哪些
2017/04/13 HTML / CSS
数控专业推荐信范文
2013/12/02 职场文书
会议活动邀请函
2014/01/27 职场文书
上课睡觉检讨书
2014/01/28 职场文书
讲文明树新风公益广告宣传方案
2014/02/25 职场文书
初中学校对照检查材料
2014/08/19 职场文书
员工工作心得体会
2019/05/07 职场文书
浅谈MySQL表空间回收的正确姿势
2021/10/05 MySQL
二维码条形码生成的JavaScript脚本库
2022/07/07 Javascript