Ele inicializador.
Se a distribuição é TRUNCATED_NORMAL, extrai amostras a partir de uma distribuição normal truncada centrado em 0
com stddev = sqrt(2 / fanIn)
onde fanIn
é o número de unidades de entrada em peso do tensor.
Se a distribuição é uniforme, extrai amostras a partir de uma distribuição uniforme dentro [-limit, limit]
, onde limit = sqrt(6 / fanIn)
(FANIN é o número de unidades de entrada em peso do tensor).
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);
NOTA:
Para um inicializador equivalente HeNormal, uso TRUNCATED_NORMAL
para o parâmetro de distribuição.
Para um inicializador equivalente HeUniform, use UNIFORM
para o parâmetro de distribuição.
Constantes
Duplo | ESCALA |
Constantes herdadas
Campos herdados
Construtores Públicos
Métodos herdados
Constantes
ESCALA dupla public static final
Construtores Públicos
public He (Ops tf, VarianceScaling.Distribution distribution, long seed)
Cria um inicializador He
Parâmetros
tf | o TensorFlow Ops |
---|---|
distribuição | O tipo de distribuição para o 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 de d. |