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]]
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]]