On inicjalizator.
Jeśli rozkład ma wartość TRUNCATED_NORMAL, pobiera próbki z obciętego rozkładu normalnego wyśrodkowanego na 0
gdzie stddev = sqrt(2 / fanIn)
gdzie fanIn
jest liczbą jednostek wejściowych w tensorze wagi.
Jeśli rozkład jest UNIFORM, pobiera próbki z rozkładu jednolitego w obrębie [-limit, limit]
, gdzie limit = sqrt(6 / fanIn)
(fanIn to liczba jednostek wejściowych w tensorze wagi).
Przykłady:
On Normalny:
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);
On jest mundurowy:
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);
NOTATKA:
W przypadku inicjatora równoważnego HeNormal użyj TRUNCATED_NORMAL
dla parametru dystrybucji.
W przypadku inicjatora równoważnego HeUniform użyj UNIFORM
dla parametru dystrybucji.
Stałe
podwójnie | SKALA |
Dziedziczone stałe
Dziedziczone pola
Konstruktorzy publiczni
Metody dziedziczone
Stałe
publiczna statyczna końcowa podwójna SKALA
Konstruktorzy publiczni
public He (Ops tf, VarianceScaling.Distribution dystrybucja, długie nasiona)
Tworzy inicjator He
Parametry
tf | operacji TensorFlow |
---|---|
dystrybucja | Typ dystrybucji dla inicjatora He. |
nasienie | materiał siewny do generowania liczb losowych. Inicjator utworzony z danego ziarna zawsze będzie generował ten sam losowy tensor dla danego kształtu i typu. |