flux tensoriel : : opérations : : Rassembler

Rassemblez les tranches des params en fonction 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

Retour:

• Output : Le tenseur de sortie.

Attributs publics

Operation operation

::tensorflow::Output output

 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
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs ValidateIndices(
  bool x
)