『TensorFlow』高級高維切片gather_nd

gather用於高級切片，有關官方文檔的介紹，關於維度的說明非常費解，示例也不太直觀，這裏給出個人解讀，示例見下面，

    indices = [[0, 0], [1, 1]]
    params = [['a', 'b'], ['c', 'd']]
    output = ['a', 'd']



    indices = [[[0, 0]], [[0, 1]]]
    params = [['a', 'b'], ['c', 'd']]
    output = [['a'], ['b']]


咱們兩次被索引對象是相同的： [2, 2]形狀，而indice的形狀各不相同，按照官網的算法計算輸出很不直觀，按照個人理解，咱們由最後一維度向外看indices便可，即 [0, 0]索引到 'a'， [1, 1]索引到 'b'，而後咱們使用索引到的對象替換掉最後一維度的索引便可，例如，對於 [[0, 0], [1, 1]]，替換掉 [0, 0]和 [1, 1]後，即爲 ['a', 'd']，而 [[[0, 0]], [[0, 1]]]替換後保留了外層兩個維度： [['a'], ['b']]。

咱們看官網另外一個例子，對應 [0, 0, 1]得到 'b0'， [1, 0, 1]得到 'b1'……，簡單替換便可。

    indices = [[[0, 0, 1], [1, 0, 1]], [[0, 1, 1], [1, 1, 0]]]
    params = [[['a0', 'b0'], ['c0', 'd0']],
           [['a1', 'b1'], ['c1', 'd1']]]
    output = [['b0', 'b1'], ['d0', 'c1']]