溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務(wù)條款》
您好，登錄后才能下訂單哦！
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]]]，這個輸出在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]]]，這個輸出在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]]]，這個輸出在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]]]，這個輸出在input的axis=0的下標(biāo)是[0, 2]，不連續(xù)，可以用tf.gather抽取。input[0]和input[2]
以上這篇tensorflow中tf.slice和tf.gather切片函數(shù)的使用就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持億速云。
免責(zé)聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點不代表本網(wǎng)站立場，如果涉及侵權(quán)請聯(lián)系站長郵箱：is@yisu.com進行舉報，并提供相關(guān)證據(jù)，一經(jīng)查實，將立刻刪除涉嫌侵權(quán)內(nèi)容。
億速云公眾號
手機網(wǎng)站二維碼