1. 程式人生 > >tf.gather()用法

tf.gather()用法

tf.gather(等待被取元素的張量,索引)

tf.gather根據索引,從輸入張量中依次取元素,構成一個新的張量。

索引的維度可以小於張量的維度。這時,取張量元素時,會把相應的低維當作一個整體取出來。

例如

假設輸入張量 [[1,2,3],[4,5,6],[7,8,9]] 是個二維的

如果只給一個一維索引0. 它就把[1,2,3]整體取出:

如果給兩個一維索引,0和1,它就形成[[1,2,3],[4,5,6]]