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)文章
Django學(xué)習(xí)教程之靜態(tài)文件的調(diào)用詳解
這篇文章主要給大家介紹了關(guān)于Django學(xué)習(xí)教程之靜態(tài)文件調(diào)用的相關(guān)資料,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家學(xué)習(xí)或者使用django具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2018-05-05在python中實(shí)現(xiàn)調(diào)用可執(zhí)行文件.exe的3種方法
今天小編就為大家分享一篇在python中實(shí)現(xiàn)調(diào)用可執(zhí)行文件.exe的3種方法,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-07-07Python中使用第三方庫(kù)xlutils來(lái)追加寫(xiě)入Excel文件示例
這篇文章主要介紹了Python中使用第三方庫(kù)xlutils來(lái)追加寫(xiě)入Excel文件示例,本文直接給出追加寫(xiě)入示例和追加效果,需要的朋友可以參考下2015-04-04pytorch神經(jīng)網(wǎng)絡(luò)從零開(kāi)始實(shí)現(xiàn)多層感知機(jī)
這篇文章主要為大家介紹了pytorch神經(jīng)網(wǎng)絡(luò)從零開(kāi)始實(shí)現(xiàn)多層感知機(jī)的示例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步2021-10-10詳解Python中pyautogui庫(kù)的最全使用方法
這篇文章主要介紹了詳解Python中pyautogui庫(kù)的最全使用方法,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-04-04python SSH模塊登錄,遠(yuǎn)程機(jī)執(zhí)行shell命令實(shí)例解析
這篇文章主要介紹了python SSH模塊登錄,遠(yuǎn)程機(jī)執(zhí)行shell命令實(shí)例解析,具有一定借鑒價(jià)值,需要的朋友可以參考下2018-01-01