tensorflow中tf.slice和tf.gather切片函數(shù)的使用
tf.slice(input_, begin, size, name=None)：按照指定的下標范圍抽取連續(xù)區(qū)域的子集
tf.gather(params, indices, validate_indices=None, name=None)：按照指定的下標集合從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]]]
假設我們要從input中抽取[[[3, 3, 3]]]，這個輸出在inputaxis=0的下標是1，axis=1的下標是0，axis=2的下標是0-2，所以begin=[1,0,0]，size=[1,1,3]。
假設我們要從input中抽取[[[3, 3, 3], [4, 4, 4]]]，這個輸出在inputaxis=0的下標是1，axis=1的下標是0-1，axis=2的下標是0-2，所以begin=[1,0,0]，size=[1,2,3]。
假設我們要從input中抽取[[[3, 3, 3], [5, 5, 5]]]，這個輸出在inputaxis=0的下標是1-2，axis=1的下標是0，axis=2的下標是0-2，所以begin=[1,0,0]，size=[2,1,3]。
假設我們要從input中抽取[[[1, 1, 1], [2, 2, 2]],[[5, 5, 5], [6, 6, 6]]]，這個輸出在input的axis=0的下標是[0, 2]，不連續(xù)，可以用tf.gather抽取。input[0]和input[2]
以上這篇tensorflow中tf.slice和tf.gather切片函數(shù)的使用就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
這篇文章主要給大家介紹了關于Django學習教程之靜態(tài)文件調用的相關資料，文中通過示例代碼介紹的非常詳細，對大家學習或者使用django具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2018-05-05
在python中實現(xiàn)調用可執(zhí)行文件.exe的3種方法
今天小編就為大家分享一篇在python中實現(xiàn)調用可執(zhí)行文件.exe的3種方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-07-07
在本篇內容里小編給大家整理的是關于python3讓print輸出不換行的方法，有需要的朋友們可以學習參考下。2020-08-08
下面小編就為大家分享一篇python獲取程序執(zhí)行文件路徑的方法(推薦)，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-04-04
這篇文章主要介紹了python判斷字符串是否純數(shù)字的方法,通過isdigit方法進行判斷,并給出了改進的實例及采用正則判斷的用法,具有一定的借鑒價值,需要的朋友可以參考下2014-11-11
Python中使用第三方庫xlutils來追加寫入Excel文件示例
這篇文章主要介紹了Python中使用第三方庫xlutils來追加寫入Excel文件示例,本文直接給出追加寫入示例和追加效果,需要的朋友可以參考下2015-04-04
這篇文章主要為大家介紹了pytorch神經網絡從零開始實現(xiàn)多層感知機的示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步2021-10-10
這篇文章主要介紹了詳解Python中pyautogui庫的最全使用方法，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2020-04-04
python SSH模塊登錄，遠程機執(zhí)行shell命令實例解析
這篇文章主要介紹了python SSH模塊登錄，遠程機執(zhí)行shell命令實例解析，具有一定借鑒價值,需要的朋友可以參考下2018-01-01
這篇文章主要介紹了Python locust工具使用詳解，幫助大家更好的理解和學習使用python，感興趣的朋友可以了解下2021-03-03