VarianceScaling

इनिशियलाइज़र अपने पैमाने को वेट टेंसर के आकार के अनुसार ढालने में सक्षम है।

distribution=TRUNCATED_NORMAL or NORMAL के साथ, नमूने शून्य के माध्य और एक मानक विचलन (छंटाई के बाद, यदि उपयोग किया जाता है) के साथ एक काटे गए/बिना काटे गए सामान्य वितरण से लिए जाते हैं stddev = Math.sqrt(scale / n) , जहां n है:

• भार टेंसर में इनपुट इकाइयों की संख्या, यदि mode=FAN_IN
• आउटपुट इकाइयों की संख्या, यदि mode=FAN_OUT
• इनपुट और आउटपुट इकाइयों की संख्या का औसत, यदि mode=FAN_AVG

distribution=UNIFORM के साथ, नमूने [-limit, limit] के भीतर एक समान वितरण से निकाले जाते हैं, जहां limit = Math.sqrt(3 * scale / n); .

उदाहरण:

      long seed = 1234l;
      float scale = 0.1f;
      VarianceScaling<TFloat32, TFloat32> initializer =
          new org.tensorflow.framework.initializers.VarianceScaling<>(
              tf, scale, Mode.FAN_IN, Distribution.UNIFORM, seed);
      Operand<TFloat32> values =
          initializer.call(tf.constant(Shape.of(2,2)), TFloat32.class);