tensorflow 用矩阵运算替换for循环 用tf.tile而不写for的方法
作者：guotong1988 发布时间：2021-08-11 11:43:55
标签：tensorflow,for,tf.tile
如下所示：
# 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]
来源：https://blog.csdn.net/guotong1988/article/details/78284560
0
投稿猜你喜欢
- 所有标准的序列操作对字符串都适用，但字符串是不可变的字符串常量：单引号：‘spa"m'双引号："spa'
在网络中传输数据时，为了防止网络拥塞，需限制流出网络的流量，使流量以比较均匀的速度向外发送，令牌桶算法就实现了这个功能， 可控制发送到网络上- 直接看下面例子my_ld = [lambda x:x*i for i in range(3)]my_list = [ld(2) for ld
- JSP 获取spring容器中bean的方法总结方案1(Web中使用)：ApplicationContext ct = WebApplica
- 这篇文章主要介绍了Python搭建HTTP服务过程图解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋
- php获取 checkbox复选框值的方法 <html xmlns="https://www.aspxhome.net/19
- 在进行浮动布局时，大多数人都深知，在必要的地方进行浮动清理：<div style="clear:both;">
- 比如有一个需求,通过sql语句,返回-5至5的随机整数.如果这一个放在PHP中,则非常简单直接用print rand(-5,5);?>
- 看到这张照片，我们一眼能够看到天宏(图中这位UED俊男)的眼睛。我们能从他的表情里读出一些他的性格。一张好的摄影作品，最重要的一点，就是这个
- 前言当使用pandas读取csv文件时，如果元素为空，则将其视为缺失值NaN（Not a Number, 非数字）。使用dropna（）方法
- 方法1：/** 功能：数据备份/恢复文件简易方法* 以日期为单位，一天一个备份文件，以当天最后备份为准* 用提交表单的形式进行操作，* 其中
- 对于大前端来说，JS可谓是我们的神器，从页面的效果到数据的传递，再到后台的业务，无处不充斥着JS的身影，但是万能的JS也有贪玩的时候，某一时
- QQWry.dat数据可以去百度搜索最新版本<％'======================================
- 这十则CSS技巧汇编于网络，作为老手已经司空见惯了，也没有什么新意，但温故而知新，或许阅读一遍也有一定的启发，本文主要面对CSS新手朋友，有
- reflow是个神奇的东西，之前Realazy说到过这个reflow，我摘出其中的重点：在CSS规范中有一个渲染对象的概念，通常用一个盒子（
- Oracle LogMiner 是Oracle公司从产品8i以后提供的一个实际非常有用的分析工具，使用该工具可以轻松获得Oracle 重作日
- 在最近一次项目有一个需求，点击按钮——异步提交——异步响应返回——根据响应返回值新开窗口。这有两个要点：异步响应之前不知道要打开窗口的URL
- 如何做一个看他爱不爱你的小测验？<％CurQ = Request.Form("CurQ")An
- 很多朋友在留言区询问关于python上传文件和字符到服务器的问题，现编针对这个给大家整理了一个解决办法。上传简单的字符串def send_s
- 最近，小明为了达成小姐姐的愿望，在某宝买到心仪的宝贝，再加上又迷上了python，就通过python轻而易举地实现了（个人声明：对Java来
