RngReadAndSkip

classe finale publique RngReadAndSkip

Avancez le compteur d’un RNG basé sur un compteur.

L'état du RNG après `rng_read_and_skip(n)` sera le même que celui après `uniform([n])` (ou toute autre distribution). L'incrément réel ajouté au compteur est un choix d'implémentation non spécifié.

Dans le cas où l'algorithme d'entrée est RNG_ALG_AUTO_SELECT, le compteur dans l'état doit être de taille int64[2], la taille maximale actuelle du compteur parmi les algorithmes. Dans ce cas, cette opération gérera le compteur comme s'il s'agissait d'un entier de 128 bits avec une disposition [lower_64bits, upper_64bits]. Si un algorithme a besoin de moins de 128 bits pour le compteur, il doit utiliser la partie gauche de int64[2]. De cette façon, l'int64[2] est compatible avec tous les algorithmes RNG actuels (Philox, ThreeFry et xla::RandomAlgorithm::RNG_DEFAULT). Les opérations RNG en aval peuvent ainsi utiliser ce compteur avec n’importe quel algorithme RNG.

Méthodes publiques

Sortie <Long>
comme Sortie ()
Renvoie le handle symbolique d'un tenseur.
RngReadAndSkip statique
créer ( Portée de portée , Opérande <?> ressource, Opérande <Integer> alg, Opérande <?> delta)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération RngReadAndSkip.
Sortie <Long>
valeur ()
L'ancienne valeur de la variable de ressource, avant l'incrémentation.

Méthodes héritées

Méthodes publiques

sortie publique <Long> asOutput ()

Renvoie le handle symbolique d'un tenseur.

Les entrées des opérations TensorFlow sont les sorties d'une autre opération TensorFlow. Cette méthode est utilisée pour obtenir un handle symbolique qui représente le calcul de l’entrée.

public static RngReadAndSkip create ( Scope scope, Operand <?> ressource, Operand <Integer> alg, Operand <?> delta)

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

Paramètres
portée portée actuelle
Ressource Handle de la variable de ressource qui stocke l’état du RNG. L'état est constitué du compteur suivi de la clé.
algue L'algorithme RNG.
delta Le montant de l'avancement.
Retour
  • une nouvelle instance de RngReadAndSkip

valeur de sortie publique <Long> ()

L'ancienne valeur de la variable de ressource, avant l'incrémentation. Étant donné que la taille de l'état dépend de l'algorithme, cette sortie sera complétée à droite par des zéros pour atteindre la forme int64[3] (la taille d'état maximale actuelle parmi les algorithmes).