tensorflow 利用expand_dims和squeeze擴展和壓縮tensor維度方式
在利用tensorflow進(jìn)行文本挖掘工作的時候，經(jīng)常涉及到維度擴展和壓縮工作。比如對文本進(jìn)行embedding操作完成之后，若要進(jìn)行卷積操作，就需要對embedded的向量擴展維度，將[batch_size, embedding_dims]擴展成為[batch_size, embedding_dims, 1]，利用tf.expand_dims(input, -1)就可實現(xiàn)，反過來用squeeze(input, -1)或者tf.squeeze（input）也可以把最第三維去掉。
tf.expand_dims()
tf.squeeze()
tf.expand_dims()
tf.expand_dims(input, axis=None, name=None, dim=None)
在第axis位置增加一個維度.
給定張量輸入，此操作在輸入形狀的維度索引軸處插入1的尺寸。 尺寸索引軸從零開始; 如果您指定軸的負(fù)數(shù)，則從最后向后計數(shù)。
如果要將批量維度添加到單個元素，則此操作非常有用。 例如，如果您有一個單一的形狀[height，width，channels]，您可以使用expand_dims（image，0）使其成為1個圖像，這將使形狀[1，高度，寬度，通道]。
例子
# 't' is a tensor of shape [2] shape(expand_dims(t, 0)) ==> [1, 2] shape(expand_dims(t, 1)) ==> [2, 1] shape(expand_dims(t, -1)) ==> [2, 1] # 't2' is a tensor of shape [2, 3, 5] shape(expand_dims(t2, 0)) ==> [1, 2, 3, 5] shape(expand_dims(t2, 2)) ==> [2, 3, 1, 5] shape(expand_dims(t2, 3)) ==> [2, 3, 5, 1]
tf.squeeze()
tf.squeeze(input, axis=None, name=None, squeeze_dims=None)
直接上例子
# 't' is a tensor of shape [1, 2, 1, 3, 1, 1] shape(squeeze(t)) ==> [2, 3] # 't' is a tensor of shape [1, 2, 1, 3, 1, 1] shape(squeeze(t, [2, 4])) ==> [1, 2, 3, 1]
以上這篇tensorflow 利用expand_dims和squeeze擴展和壓縮tensor維度方式就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
pytorch 實現(xiàn)多個Dataloader同時訓(xùn)練
這篇文章主要介紹了pytorch 實現(xiàn)多個Dataloader同時訓(xùn)練的操作，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2021-05-05
這篇文章主要為大家詳細(xì)介紹了python郵件發(fā)送smtplib的使用方法，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-05-05
python3 實現(xiàn)的對象與json相互轉(zhuǎn)換操作示例
這篇文章主要介紹了python3 實現(xiàn)的對象與json相互轉(zhuǎn)換操作,結(jié)合實例形式分析了Python3使用json模塊針對json格式數(shù)據(jù)轉(zhuǎn)換操作的相關(guān)實現(xiàn)技巧,需要的朋友可以參考下2019-08-08
這篇文章主要介紹了Python與C語言分別完成排序的實例，在Python與C語言基本類型的排序中特別有用,下面我們一起進(jìn)入文章學(xué)習(xí)更詳細(xì)的內(nèi)容吧,需要的朋友可以參考下2022-03-03
python中內(nèi)置庫os與sys模塊的詳細(xì)介紹
這篇文章主要介紹了python中內(nèi)置庫os與sys模塊的詳細(xì)介紹，文章圍繞主題展開詳細(xì)的內(nèi)容介紹，具有一定的參考價值，需要的朋友可以參考一下2022-08-08
python利用openpyxl拆分多個工作表的工作簿的方法
這篇文章主要介紹了python利用openpyxl拆分多個工作表的工作簿的方法，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-09-09
這篇文章主要介紹一個可以將音頻提升音量的python常用工具，代碼具有一定的學(xué)習(xí)價值，感興趣的小伙伴可以了解一下2021-11-11
在python中利用pycharm自定義代碼塊教程(三步搞定)
這篇文章主要介紹了在python中利用pycharm自定義代碼塊教程(三步搞定)，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-04-04
python使用selenium登錄QQ郵箱（附帶滑動解鎖）
這篇文章主要為大家詳細(xì)介紹了python使用selenium登錄QQ郵箱，帶滑動解鎖登錄功能，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-01-01
Python實現(xiàn)在Word中創(chuàng)建表格并填入數(shù)據(jù)與圖片
在Word中,表格是一個強大的工具，本文主要為大家介紹了如何使用Python在Word中創(chuàng)建表格并填入數(shù)據(jù)、圖片,以及設(shè)置表格樣式等,感興趣的可以了解下2024-03-03