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学习数据结构实例代码
May 11 Python
Python计算三角函数之asin()方法的使用
May 15 Python
python获取指定路径下所有指定后缀文件的方法
May 26 Python
Python脚本暴力破解栅栏密码
Oct 19 Python
使用python实现接口的方法
Jul 07 Python
mac安装pytorch及系统的numpy更新方法
Jul 26 Python
python二维码操作:对QRCode和MyQR入门详解
Jun 24 Python
python使用socket实现的传输demo示例【基于TCP协议】
Sep 24 Python
python GUI库图形界面开发之PyQt5中QWebEngineView内嵌网页与Python的数据交互传参详细方法实例
Feb 26 Python
Python使用多进程运行含有任意个参数的函数
May 02 Python
详解Python 3.10 中的新功能和变化
Apr 28 Python
Python3接口性能测试实例代码
Jun 20 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数据库连接类~~做成一个分页类!
2006/11/25 PHP
How do I change MySQL timezone?
2008/03/26 PHP
php中将一个对象保存到Session中的方法
2015/03/13 PHP
PHP中trait使用方法详细介绍
2017/05/21 PHP
深入解析PHP底层机制及相关原理
2020/12/11 PHP
Javascript valueOf 使用方法
2008/12/28 Javascript
js获取height和width的方法说明
2013/01/06 Javascript
JS实现下拉框的动态添加(附效果)
2013/04/03 Javascript
Jquery在指定DIV加载HTML示例代码
2014/02/17 Javascript
jQuery 移动端artEditor富文本编辑器
2016/01/11 Javascript
基于JavaScript实现TAB标签效果
2016/01/12 Javascript
Node.js 实现简单小说爬虫实例
2016/11/18 Javascript
JS返回只包含数字类型的数组实例分析
2016/12/16 Javascript
JavaScript实现左右下拉框动态增删示例
2017/03/09 Javascript
利用Dectorator分模块存储Vuex状态的实现
2019/02/05 Javascript
Vue源码探究之虚拟节点的实现
2019/04/17 Javascript
详解Vue项目中实现锚点定位
2019/04/24 Javascript
Layui table field初始化加载时进行隐藏的方法
2019/09/19 Javascript
微信小程序indexOf的替换方法(推荐)
2020/01/14 Javascript
一文秒懂nodejs中的异步编程
2021/01/28 NodeJs
pandas DataFrame数据转为list的方法
2018/04/11 Python
python 读取.csv文件数据到数组(矩阵)的实例讲解
2018/06/14 Python
Win10下python3.5和python2.7环境变量配置教程
2018/09/18 Python
Python使用正则实现计算字符串算式
2019/12/29 Python
Pytorch中的VGG实现修改最后一层FC
2020/01/15 Python
matplotlib quiver箭图绘制案例
2020/04/17 Python
CSS3 Pie工具推荐--让IE6-8支持一些优秀的CSS3特性
2014/09/02 HTML / CSS
全球最大的中文旅行网站:去哪儿网
2017/11/16 全球购物
美国宠物护理专家:Revival Animal Health
2020/01/05 全球购物
Columbia Sportswear法国官网:全球户外品牌
2020/09/25 全球购物
年度考核自我鉴定
2014/02/02 职场文书
环保倡议书
2014/04/14 职场文书
三严三实对照检查材料
2014/08/25 职场文书
2014年最新领导班子整改方案
2014/09/27 职场文书
2014年化妆品销售工作总结
2014/12/01 职场文书
TypeScript中条件类型精读与实践记录
2021/10/05 Javascript