tensorflow 用矩陣運算替換for循環(huán) 用tf.tile而不寫for的方法
更新時間：2018年07月27日 09:56:39 作者：guotong1988
今天小編就為大家分享一篇tensorflow 用矩陣運算替換for循環(huán) 用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實現(xiàn)更高速的版本
# 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循環(huán) 用tf.tile而不寫for的方法就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了Python中replace方法,有一定的實用價值,需要的朋友可以參考下2014-08-08
Python如何運用pyaudio庫去做一個固定采樣率音頻錄制器
這篇文章主要介紹了Python如何運用pyaudio庫去做一個固定采樣率音頻錄制器問題，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2023-05-05
這篇文章主要介紹了教你用Python寫一個植物大戰(zhàn)僵尸小游戲,文中有非常詳細的代碼示例,對正在學(xué)習(xí)python的小伙伴們有非常好的幫助,需要的朋友可以參考下2021-04-04
這篇文章主要介紹了PyCharm2019安裝教程(圖文教程)，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-09-09
這篇文章主要介紹了使用Python讀寫文本文件及編寫簡單的文本編輯器的方法,可以利用簡單的Python代碼實現(xiàn)編輯器中的讀取清空和文本文件的新建等操作,需要的朋友可以參考下2016-03-03
這篇文章主要介紹了GraphQL在Django中的使用教程,本文結(jié)合示例代碼給大家介紹的非常詳細，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2022-12-12
python 定時器,實現(xiàn)每天凌晨3點執(zhí)行的方法
今天小編就為大家分享一篇python 定時器,實現(xiàn)每天凌晨3點執(zhí)行的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-02-02
這篇文章主要介紹了查看python安裝路徑及pip安裝的包列表及路徑,本文給大家介紹的非常詳細，具有一定的參考借鑒價值，需要的朋友可以參考下2019-04-04
這篇文章主要為大家介紹了python?skimage圖像處理，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2022-06-06
這篇文章主要介紹了詳解如何管理多個Python版本和虛擬環(huán)境，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2019-05-05