tensorflow中tf.slice和tf.gather切片函數(shù)的使用
tf.slice(input_, begin, size, name=None)：按照指定的下標(biāo)范圍抽取連續(xù)區(qū)域的子集
tf.gather(params, indices, validate_indices=None, name=None)：按照指定的下標(biāo)集合從axis=0中抽取子集，適合抽取不連續(xù)區(qū)域的子集
輸出：
input = [[[1, 1, 1], [2, 2, 2]],
[[3, 3, 3], [4, 4, 4]],
[[5, 5, 5], [6, 6, 6]]]
tf.slice(input, [1, 0, 0], [1, 1, 3]) ==> [[[3, 3, 3]]]
tf.slice(input, [1, 0, 0], [1, 2, 3]) ==> [[[3, 3, 3],
[4, 4, 4]]]
tf.slice(input, [1, 0, 0], [2, 1, 3]) ==> [[[3, 3, 3]],
[[5, 5, 5]]]
tf.gather(input, [0, 2]) ==> [[[1, 1, 1], [2, 2, 2]],
[[5, 5, 5], [6, 6, 6]]]
假設(shè)我們要從input中抽取[[[3, 3, 3]]]，這個(gè)輸出在inputaxis=0的下標(biāo)是1，axis=1的下標(biāo)是0，axis=2的下標(biāo)是0-2，所以begin=[1,0,0]，size=[1,1,3]。
假設(shè)我們要從input中抽取[[[3, 3, 3], [4, 4, 4]]]，這個(gè)輸出在inputaxis=0的下標(biāo)是1，axis=1的下標(biāo)是0-1，axis=2的下標(biāo)是0-2，所以begin=[1,0,0]，size=[1,2,3]。
假設(shè)我們要從input中抽取[[[3, 3, 3], [5, 5, 5]]]，這個(gè)輸出在inputaxis=0的下標(biāo)是1-2，axis=1的下標(biāo)是0，axis=2的下標(biāo)是0-2，所以begin=[1,0,0]，size=[2,1,3]。
假設(shè)我們要從input中抽取[[[1, 1, 1], [2, 2, 2]],[[5, 5, 5], [6, 6, 6]]]，這個(gè)輸出在input的axis=0的下標(biāo)是[0, 2]，不連續(xù)，可以用tf.gather抽取。input[0]和input[2]
以上這篇tensorflow中tf.slice和tf.gather切片函數(shù)的使用就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
詳解windows python3.7安裝numpy問(wèn)題的解決方法
這篇文章主要介紹了windows python3.7安裝numpy問(wèn)題的解決方法，小編覺(jué)得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2018-08-08
這篇文章主要介紹了Python的random模塊的相關(guān)內(nèi)容，具有一定借鑒價(jià)值，需要的朋友可以參考下。2017-12-12
Pandas 是一個(gè)用于數(shù)據(jù)分析的 Python 第三方庫(kù)，能夠處理和分析不同格式的數(shù)據(jù)，Pandas 提供了兩種數(shù)據(jù)結(jié)構(gòu)，分別為 Series 和 DataFrame，靈活而方便地進(jìn)行數(shù)據(jù)分析和操作，本文通過(guò)實(shí)例代碼給大家介紹的非常詳細(xì)，需要的朋友參考下吧2023-06-06
利用Python產(chǎn)生加密表和解密表的實(shí)現(xiàn)方法
這篇文章主要介紹了利用Python產(chǎn)生加密表和解密表的實(shí)現(xiàn)方法，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2019-10-10
這篇文章主要介紹了pycharm2020.2 配置使用的方法，本文通過(guò)圖文并茂的形式給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2020-09-09
Python根據(jù)服務(wù)獲取端口號(hào)的方法
這篇文章主要介紹了Python根據(jù)服務(wù)獲取端口號(hào),文中給大家提到了linux查看端口開(kāi)啟端口的方法，需要的朋友可以參考下2019-09-09
python模擬實(shí)現(xiàn)斗地主發(fā)牌
這篇文章主要為大家詳細(xì)介紹了python代碼模擬實(shí)現(xiàn)斗地主發(fā)牌，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2020-01-01
這篇文章主要介紹了Mac安裝python3的方法步驟，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2019-08-08
今天小編就為大家分享一篇python 獲取微信好友列表的方法(微信web)，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-02-02
Python利用atexit模塊實(shí)現(xiàn)優(yōu)雅處理程序退出
Python的atexit模塊提供了一種方便的方式來(lái)注冊(cè)這些退出時(shí)執(zhí)行的函數(shù),文中的示例代碼講解詳細(xì),感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下2024-03-03