Il initialise.
Si la distribution est TRUNCATED_NORMAL, elle tire des échantillons d'une distribution normale tronquée centrée sur 0
avec stddev = sqrt(2 / fanIn)
où fanIn
est le nombre d'unités d'entrée dans le tenseur de poids.
Si la distribution est UNIFORME, elle tire des échantillons d'une distribution uniforme dans [-limit, limit]
, où limit = sqrt(6 / fanIn)
(fanIn est le nombre d'unités d'entrée dans le tenseur de poids).
Exemples :
Il est normal :
long seed = 1001l; He<TFloat32, TFloat32> initializer = new org.tensorflow.framework.initializers.He<>(tf, Distribution.TRUNCATED_NORMAL, seed);); Operand<TFloat32> values = initializer.call(tf.constant(Shape.of(2,2)), TFloat32.class);
L'uniforme :
long seed = 1001l; He<TFloat32, TFloat32> initializer = new org.tensorflow.framework.initializers.He<>(tf, Distribution.UNIFORM, seed);); Operand<TFloat32> values = initializer.call(tf.constant(Shape.of(2,2)), TFloat32.class);
NOTE:
Pour un initialiseur équivalent HeNormal, utilisez TRUNCATED_NORMAL
pour le paramètre de distribution.
Pour un initialiseur équivalent HeUniform, utilisez UNIFORM
pour le paramètre de distribution.
Constantes
double | ÉCHELLE |
Constantes héritées
Champs hérités
Constructeurs Publics
Méthodes héritées
Constantes
public statique final double SCALE
Constructeurs Publics
public He (Ops tf, VarianceScaling.Distribution distribution, graine longue)
Crée un initialiseur He
Paramètres
tf | les opérations TensorFlow |
---|---|
distribution | Le type de distribution pour l’initialiseur He. |
graine | la graine pour la génération de nombres aléatoires. Un initialiseur créé avec une graine donnée produira toujours le même tenseur aléatoire pour une forme et un type donnés. |