パブリッククラス彼
彼は初期化子です。
分布が TRUNCATED_NORMAL の場合、 stddev = sqrt(2 / fanIn)
で0
を中心とする切り捨て正規分布からサンプルが抽出されます。ここで、 fanIn
重みテンソルの入力単位の数です。
分布が UNIFORM の場合、 [-limit, limit]
内の一様分布からサンプルが抽出されます。ここで、 limit = sqrt(6 / fanIn)
(fanIn は重みテンソルの入力単位の数です)。
例:
彼は普通です:
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);
彼のユニフォーム:
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);
注記:
HeNormal と同等の初期化子の場合は、分布パラメーターにTRUNCATED_NORMAL
を使用します。
HeUniform と同等のイニシャライザの場合は、分布パラメータにUNIFORM
を使用します。
定数
ダブル | 規模 |
継承された定数
継承されたフィールド
パブリックコンストラクター
継承されたメソッド
定数
パブリック静的最終倍精度スケール
定数値: 2.0
パブリックコンストラクター
public He (Ops tf、 VarianceScaling.Distribution分布、ロングシード)
He イニシャライザを作成します
パラメータ
TF | TensorFlow オペレーション |
---|---|
分布 | He イニシャライザの分散タイプ。 |
シード | 乱数生成のシード。特定のシードを使用して作成されたイニシャライザは、常に特定のシェイプと dtype に対して同じランダム テンソルを生成します。 |