Cette page a été traduite par l'API Cloud Translation.
Switch to English

tensorflow :: ops :: GatherNd

#include <array_ops.h>

Rassembler les tranches de params dans une Tensor avec la forme spécifiée par indices .

Résumé

indices est un tenseur de nombre entier K dimensions, mieux considéré comme un dimensionnel (K-1) tenseur des indices dans params , où chaque élément définit une tranche de params :

output[\\(i_0, ..., i_{K-2}\\)] = params[indices[\\(i_0, ..., i_{K-2}\\)]]

Alors que dans tf.gather indices définit les tranches dans l' axis dimension de params , dans tf.gather_nd , les indices définit les tranches dans la première N dimensions de params , où N = indices.shape[-1] .

La dernière dimension des indices peut être au plus le rang de params :

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

La dernière dimension des indices correspond à des éléments (si indices.shape[-1] == params.rank ) ou des tranches (si indices.shape[-1] < params.rank ) le long de la dimension indices.shape[-1] de params . Le tenseur de sortie a la forme

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

Notez que le CPU, si un indice de limite se trouve, une erreur est renvoyée. Le GPU, si un index de limite se trouve, un 0 est stockée dans la valeur de sortie correspondante.

Quelques exemples ci-dessous.

indexation simple dans une matrice:

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

Tranche indexation dans une matrice:

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

L'indexation dans un 3-tenseur:

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

l'indexation par lots dans une matrice:

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

L'indexation par lots de la tranche dans une matrice:

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

l'indexation par lots dans un 3-tenseur:

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

Voir aussi tf.gather et tf.batch_gather .

Arguments:

  • portée: A Portée objet
  • params: Le tenseur à partir desquels les valeurs.
  • indices: Indice tenseur.

Retour:

  • Output : Les valeurs de params recueillies à partir des indices donnés par des indices , avec la forme indices.shape[:-1] + params.shape[indices.shape[-1]:] .

Constructeurs et Destructeurs

GatherNd (const :: tensorflow::Scope & scope, :: tensorflow::Input params, :: tensorflow::Input indices)

attributs publics

operation
output

Les fonctions publiques

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

attributs publics

opération

 Operation operation

production

:: tensorflow::Output output

Les fonctions publiques

GatherNd

 GatherNd(
  const :: tensorflow::Scope & scope,
  :: tensorflow::Input params,
  :: tensorflow::Input indices
)

nœud

::tensorflow::Node * node() const 

opérateur :: tensorflow :: entrée

 operator::tensorflow::Input() const 

opérateur :: tensorflow :: sortie

 operator::tensorflow::Output() const