klasa publiczna On
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 Mundur:
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
Wartość stała: 2,0
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. |
| nasionko | 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. |