Une opération divise le tuple XLA des données de déduplication d’entrée en tenseurs entiers et à virgule flottante.
Les données de déduplication sont un tuple XLA composé de valeurs entières et à virgule flottante. Cette opération consiste à diviser ces valeurs en deux groupes pour deux types et à construire chaque groupe comme un tenseur à renvoyer.
Classes imbriquées
classe | SplitDedupData.Options | Attributs facultatifs pour SplitDedupData |
Méthodes publiques
SplitDedupData.Options statique | config (configuration de chaîne) |
static <T étend le nombre, U étend le nombre> SplitDedupData <T, U> | créer ( Scope scope, Operand <?> input, Class<T> integerType, Class<U> floatType, String tupleMask, Options... options) Méthode d'usine pour créer une classe encapsulant une nouvelle opération SplitDedupData. |
Sortie <U> | floatTensor () Un tenseur flottant 1D comprend des éléments flottants du tuple de données de déduplication. |
Sortie <T> | Tenseur entier () Un tenseur entier 1D comprend des éléments entiers du tuple de données de déduplication. |
Méthodes héritées
Méthodes publiques
public static SplitDedupData <T, U> create ( Scope scope, Operand <?> input, Class<T> integerType, Class<U> floatType, String tupleMask, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération SplitDedupData.
Paramètres
portée | portée actuelle |
---|---|
saisir | Un tuple XLA comprenant des éléments entiers et flottants comme tuple de données de déduplication. |
Type entier | type tenseur_entier. Types autorisés : int32, int64, uint32, uint64. |
TypeFlottant | type float_tensor. Types autorisés : half, bfloat16, float. |
tupleMasque | Une chaîne TensorProto sérialisée de masque de tuple de sortie. Ce masque est un tenseur 2D, avec la première colonne comme type d'élément tuple et la deuxième colonne comme étendue de ce type. Par exemple, un tuple de sortie de (1, 2, 0.1, 3), son masque est [[0, 2], [1, 1], [0, 1]]. Nous n'attendons que deux types d'éléments : integer(0) et float(1). |
choix | porte des valeurs d'attributs facultatifs |
Retour
- une nouvelle instance de SplitDedupData
sortie publique <U> floatTensor ()
Un tenseur flottant 1D comprend des éléments flottants du tuple de données de déduplication.
sortie publique <T> entierTensor ()
Un tenseur entier 1D comprend des éléments entiers du tuple de données de déduplication.