tensorflow 利用expand_dims和squeeze擴展和壓縮tensor維度方式
更新時間：2020年02月07日 10:13:18 作者：LauJames
今天小編就為大家分享一篇tensorflow 利用expand_dims和squeeze擴展和壓縮tensor維度方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
在利用tensorflow進行文本挖掘工作的時候，經常涉及到維度擴展和壓縮工作。比如對文本進行embedding操作完成之后，若要進行卷積操作，就需要對embedded的向量擴展維度，將[batch_size, embedding_dims]擴展成為[batch_size, embedding_dims, 1]，利用tf.expand_dims(input, -1)就可實現，反過來用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的尺寸。 尺寸索引軸從零開始; 如果您指定軸的負數，則從最后向后計數。
如果要將批量維度添加到單個元素，則此操作非常有用。 例如，如果您有一個單一的形狀[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維度方式就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
這篇文章主要介紹了關于python3中setup.py小概念解析,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2019-08-08
這篇文章主要介紹了分別利用PyQT5和Tkinter實現電腦的定時關機小程序，文中的示例代碼講解詳細，對我們學習Python有一定的幫助，快跟隨小編一起學習一下吧2021-12-12
今天小編就為大家分享一篇關于Tensorflow分布式并行策略，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-02-02
這篇文章主要給大家介紹了關于Python中typing模塊與類型注解的使用方法，文中通過示例代碼介紹的非常詳細，對大家學習或者使用Python具有一定的參考學習價值，需要的朋友們下面來一起學習學習吧2019-08-08
DjangoRestFramework 使用 simpleJWT 登陸認證完整記錄
Djangorestframework-simplejwt是Django REST Framework框架的一個jwt插件，使用 python http 工具進行接口測試的方法文中給大家提到，重點給大家分享djangorestframework-simplejwt 使用記錄及登陸認證的完成過程，感興趣的朋友跟隨小編一起看看吧2021-06-06
本篇文章主要介紹了python分布式環(huán)境下的限流器的示例，小編覺得挺不錯的，現在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2017-10-10
這篇文章主要介紹了Python實現利用最大公約數求三個正整數的最小公倍數,涉及Python數學運算相關操作技巧,需要的朋友可以參考下2017-09-09
這篇文章主要介紹了Python批量將圖片灰度化的實現代碼，本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值,需要的朋友可以參考下2020-04-04
在本篇文章里小編給大家整理的是一篇關于python關鍵字傳遞參數實例分析內容，有需要的朋友們可以學習參考下。2021-06-06
解決django-xadmin列表頁filter關聯對象搜索問題
今天小編就為大家分享一篇解決django-xadmin列表頁filter關聯對象搜索問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-11-11