flux tensoriel : : opérations : : Rassembler

#include <array_ops.h>

Rassemblez les tranches des params en fonction des indices .

Résumé

indices doivent être un tenseur entier de n'importe quelle dimension (généralement 0-D ou 1-D). Produit un tenseur de sortie avec la forme indices.shape + params.shape[1:] où : 

    # Scalar indices
    output[:, ..., :] = params[indices, :, ... :]

    # Vector indices
    output[i, :, ..., :] = params[indices[i], :, ... :]

    # Higher rank indices
    output[i, ..., j, :, ... :] = params[indices[i, ..., j], :, ..., :]

Si indices sont une permutation et len(indices) == params.shape[0] alors cette opération permutera params en conséquence.

validate_indices : obsolète. Si cette opération est affectée au processeur, les valeurs des indices sont toujours validées pour être dans la plage. S'ils sont attribués au GPU, les index hors limites entraînent un comportement sûr mais non spécifié, qui peut inclure la génération d'une erreur.

Arguments :

  • scope : un objet Scope

Retours :

  • Output : Le tenseur de sortie.

Constructeurs et Destructeurs

Gather (const :: tensorflow::Scope & scope, :: tensorflow::Input params, :: tensorflow::Input indices)
Gather (const :: tensorflow::Scope & scope, :: tensorflow::Input params, :: tensorflow::Input indices, const Gather::Attrs & attrs)

Attributs publics

operation
output

Fonctions publiques

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

Fonctions statiques publiques

ValidateIndices (bool x)

Structures

tensorflow :: ops :: Gather :: Attrs

Setters d'attributs facultatifs pour Gather .

Attributs publics

opération 

Operation operation

sortir 

::tensorflow::Output output

Fonctions publiques

Rassembler 

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

Rassembler 

 Gather(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input params,
  ::tensorflow::Input indices,
  const Gather::Attrs & attrs
)

nœud 

::tensorflow::Node * node() const

opérateur :: tensorflow :: Entrée 

 operator::tensorflow::Input() const

opérateur :: tensorflow :: Sortie 

 operator::tensorflow::Output() const

Fonctions statiques publiques

ValiderIndices 

Attrs ValidateIndices(
  bool x
)