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
)