tensorflow :: opérations :: Rassembler

#include <array_ops.h>

Rassemblez les tranches des params 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 shape 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 est une permutation et len(indices) == params.shape[0] alors cette opération permutera les params conséquence.

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

Arguments:

Retour:

  • 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)

Structs

tensorflow :: ops :: Gather :: Attrs

Définisseurs d'attributs facultatifs pourGather .

Attributs publics

opération

Operation operation

production

::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 

operator :: tensorflow :: Input

 operator::tensorflow::Input() const 

operator :: tensorflow :: Output

 operator::tensorflow::Output() const 

Fonctions statiques publiques

ValiderIndices

Attrs ValidateIndices(
  bool x
)