tensorflow::ops::Gather

Rassemblez les tranches de params en fonction indices .

Résumé

indices doivent être un tenseur d'entiers de dimension quelconque (généralement 0D ou 1D). Produit un tenseur de sortie de 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 est une permutation et len(indices) == params.shape[0] alors cette opération permutera params en conséquence.

validate_indices : DÉPRÉCIÉ. Si cette opération est exécutée sur le processeur, les valeurs des indices sont toujours vérifiées afin de s'assurer qu'elles sont comprises dans la plage autorisée. Si elle est exécutée sur le GPU, les indices hors limites entraînent un comportement sûr mais non spécifié, pouvant inclure la levée d'une erreur.

Arguments :

• portée : un objet Scope

Retours :

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