パブリッククラス彼
彼は初期化子です。
分布が 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 に対して同じランダム テンソルを生成します。 |