RaggedGather

classe finale publique RaggedGather

Rassemblez des tranches irrégulières de l'axe `params` `0` en fonction des `indices`.

Génère une sortie `RaggedTensor` composée de `output_dense_values` et `output_nested_splits`, telle que :

output.shape = indices.shape + params.shape[1:]
 output.ragged_rank = indices.shape.ndims + params.ragged_rank
 output[i...j, d0...dn] = params[indices[i...j], d0...dn]
où
  • `params = ragged.from_nested_row_splits(params_dense_values, params_nested_splits)` fournit les valeurs qui doivent être collectées.
  • `indices` est un tenseur dense de type `int32` ou `int64`, indiquant quelles valeurs doivent être collectées.
  • `output = ragged.from_nested_row_splits(output_dense_values, output_nested_splits)` est le tenseur de sortie.
(Remarque : cette opération C++ est utilisée pour implémenter l'opération Python `tf.ragged.gather` de niveau supérieur, qui prend également en charge les indices irréguliers.)

Constantes

Chaîne OP_NAME Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Méthodes publiques

static <T étend TNumber , U étend TType > RaggedGather <T, U>
créer ( Scope scope, Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, Operand <? extends TNumber > indices, Long OUTPUTRAGGEDRANK)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération RaggedGather.
Sortie <U>
valeurs de sortieDense ()
Les `flat_values` pour le RaggedTensor renvoyé.
Liste< Sortie <T>>
sortieNestedSplits ()
Les tenseurs `nested_row_splits` qui définissent le partitionnement des lignes pour le RaggedTensor renvoyé.

Méthodes héritées

Constantes

chaîne finale statique publique OP_NAME

Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Valeur constante : "RaggedGather"

Méthodes publiques

public static RaggedGather <T, U> créer ( Scope scope, Iterable < Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, Operand <? extends TNumber > indices, Long OUTPUTRAGGEDRANK)

Méthode d'usine pour créer une classe encapsulant une nouvelle opération RaggedGather.

Paramètres
portée portée actuelle
paramsNestedSplits Les tenseurs `nested_row_splits` qui définissent le partitionnement des lignes pour l'entrée RaggedTensor `params`.
paramsDenseValues Les `flat_values` pour le RaggedTensor `params`. Il y a eu un changement de terminologie au niveau python de dense_values ​​à flat_values, donc dense_values ​​est le nom obsolète.
indices Indices dans la dimension la plus externe des « paramètres » des valeurs qui doivent être collectées.
SORTIEGEDRANK Le rang irrégulier de la sortie RaggedTensor. `output_nested_splits` contiendra ce nombre de tenseurs `row_splits`. Cette valeur doit être égale à « indices.shape.ndims + params.ragged_rank - 1 ».
Retours
  • une nouvelle instance de RaggedGather

sortie publique <U> sortieDenseValues ​​()

Les `flat_values` pour le RaggedTensor renvoyé.

liste publique < Sortie <T>> sortieNestedSplits ()

Les tenseurs `nested_row_splits` qui définissent le partitionnement des lignes pour le RaggedTensor renvoyé.