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.
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> | |
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
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
liste publique < Sortie <T>> sortieNestedSplits ()
Les tenseurs `nested_row_splits` qui définissent le partitionnement des lignes pour le RaggedTensor renvoyé.