flux tenseur :: opérations :: Rassembler

Rassemblez les tranches de params selon les indices .

Sommaire

les 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 params en conséquence.

validate_indices : DÉCONSEILLÉ. 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 affectés au GPU, les index hors limites entraînent un comportement sûr mais non spécifié, qui peut inclure le déclenchement 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
)