今天小编就为大家分享一篇tensorflow 用矩阵运算替换for循环 用tf.tile而不写for的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
如下所示：
# 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]<i style="color:transparent">来源gaodai$ma#com搞$$代**码网</i>*[32,400,30,200]
以上就是tensorflow 用矩阵运算替换for循环 用tf.tile而不写for的方法的详细内容，更多请关注gaodaima搞代码网其它相关文章！
搞代码网（gaodaima.com）提供的所有资源部分来自互联网，如果有侵犯您的版权或其他权益，请说明详细缘由并提供版权或权益证明然后发送到邮箱[email protected]，我们会在看到邮件的第一时间内为您处理，或直接联系QQ：872152909。本网站采用BY-NC-SA协议进行授权
转载请注明原文链接：tensorflow 用矩阵运算替换for循环 用tf.tile而不写for的方法
转载请注明原文链接：tensorflow 用矩阵运算替换for循环 用tf.tile而不写for的方法
