GatherNd

`params`의 조각을 `indices`로 지정된 모양의 Tensor로 수집합니다.

'indices'는 K차원 정수 텐서이며, 'params'에 대한 인덱스의 (K-1)차원 텐서로 가장 잘 생각됩니다. 여기서 각 요소는 'params' 조각을 정의합니다.

출력[\\(i_0, ..., i_{K-2}\\)] = params[인덱스[\\(i_0, ..., i_{K-2}\\)]]

tf.gather 에서 `indices`는 `params`의 `axis` 차원으로 슬라이스를 정의하는 반면, tf.gather_nd 에서 `indices`는 `params`의 첫 번째 `N` 차원으로 슬라이스를 정의합니다. 여기서 `N = indices.shape [-1]`.

`indices`의 마지막 차원은 기껏해야 `params`의 순위일 수 있습니다.

indices.shape[-1] <= params.rank

`indices`의 마지막 차원은 `indices.shape[-1] == params.rank`인 경우) 차원 또는 `indices.shape[-1] < params.rank`인 경우 요소에 해당합니다. `params`의 모양[-1]`. 출력 텐서에는 모양이 있습니다.

indices.shape[:-1] + params.shape[indices.shape[-1]:]

CPU에서 범위를 벗어난 인덱스가 발견되면 오류가 반환됩니다. GPU에서는 범위를 벗어난 인덱스가 발견되면 해당 출력 값에 0이 저장됩니다.

아래에 몇 가지 예가 있습니다.

단순 행렬 인덱싱:

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

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

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

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

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

indices = [[[1]], [[0]]]
     params = [[['a0', 'b0'], ['c0', 'd0']],
               [['a1', 'b1'], ['c1', 'd1']]]
     output = [[[['a1', 'b1'], ['c1', 'd1']]],
               [[['a0', 'b0'], ['c0', 'd0']]]]
 
     indices = [[[0, 1], [1, 0]], [[0, 0], [1, 1]]]
     params = [[['a0', 'b0'], ['c0', 'd0']],
               [['a1', 'b1'], ['c1', 'd1']]]
     output = [[['c0', 'd0'], ['a1', 'b1']],
               [['a0', 'b0'], ['c1', 'd1']]]
 
 
     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']]