El inicializador.
Si la distribución es TRUNCATED_NORMAL, señala a muestras de una distribución normal truncada centrada en 0 con stddev = sqrt(2 / fanIn) donde fanIn es el número de unidades de entrada en el tensor de peso.
Si la distribución es uniforme, se basa muestras de una distribución uniforme dentro de [-limit, limit] , donde limit = sqrt(6 / fanIn) (Fanin es el número de unidades de entrada en el tensor de peso).
Ejemplos:
Él 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);
NOTA:
Para un inicializador equivalente HeNormal, el uso TRUNCATED_NORMAL para el parámetro de distribución.
Para un inicializador equivalente HeUniform, utilice UNIFORM para el parámetro de distribución.
Constantes
| doble | ESCALA |
Constantes heredadas
Campos heredados
Constructores públicos
Métodos heredados
Constantes
Doble escala public static final
Constructores públicos
Él pública (OPS tf, VarianceScaling.Distribution distribución, larga semilla)
Crea un inicializador He
Parámetros
| tf | las operaciones de TensorFlow |
|---|---|
| distribución | El tipo de distribución del inicializador He. |
| semilla | la semilla para la generación de números aleatorios. Un inicializador creado con una semilla dada siempre producirá el mismo tensor aleatorio para una forma y tipo d dados. |