classe pública Ele
Ele inicializador.
Se a distribuição for TRUNCATED_NORMAL, ela extrai amostras de uma distribuição normal truncada centrada em 0 com stddev = sqrt(2 / fanIn) onde fanIn é o número de unidades de entrada no tensor de peso.
Se a distribuição for UNIFORM, ela extrai amostras de uma distribuição uniforme dentro de [-limit, limit] , onde limit = sqrt(6 / fanIn) (fanIn é o número de unidades de entrada no tensor de peso).
Exemplos:
Ele 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);
Ele 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);
OBSERVAÇÃO:
Para um inicializador equivalente HeNormal, use TRUNCATED_NORMAL para o parâmetro de distribuição.
Para um inicializador equivalente HeUniform, use UNIFORM para o parâmetro de distribuição.
Constantes
| dobro | ESCALA |
Constantes herdadas
Campos Herdados
Construtores Públicos
Métodos herdados
Constantes
público estático final duplo ESCALA
Valor Constante: 2,0
Construtores Públicos
public He (Ops tf, distribuição VarianceScaling.Distribution , seed longo)
Cria um inicializador He
Parâmetros
| TF | as operações do TensorFlow |
|---|---|
| distribuição | O tipo de distribuição do inicializador He. |
| semente | a semente para geração de números aleatórios. Um inicializador criado com uma determinada semente sempre produzirá o mesmo tensor aleatório para uma determinada forma e tipo. |