RaggedGather

classe finale publique RaggedGather

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

Produit 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.)

Méthodes publiques

statique <T étend le nombre, U, V étend le nombre> RaggedGather <T, U>	créer ( Portée de portée , Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, Operand <V> 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

De la classe org.tensorflow.op.PrimitiveOp

booléen final	est égal (Obj objet)
entier final	Code de hachage ()
Opération	op () Renvoie l' Operation sous-jacente
Chaîne finale	àChaîne ()

De la classe java.lang.Object

booléen	est égal à (Objet arg0)
Classe finale<?>	obtenirClasse ()
int	Code de hachage ()
vide final	notifier ()
vide final	notifierTous ()
Chaîne	àChaîne ()
vide final	attendre (long arg0, int arg1)
vide final	attendez (long arg0)
vide final	attendez ()

Méthodes publiques

public static RaggedGather <T, U> créer ( Scope scope, Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, Operand <V> 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 ».

Retour

• 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é.