docstr: Kaiming normal gain.
docstr_tok: ["Kaiming","normal","gain","."]
code: def test_kaiming_normal_gain(self):
      """Kaiming normal gain."""
      shapes = (
          (4, 3, 2, 10000),
          (12, 2, 10000),
          (24, 10000),
          (32, 10000),
      )
      expected_gain = (0.20412415, 0.20412415, 0.20412415, 0.17677670)
      for s, egain in zip(shapes, expected_gain):
          gain = objax.nn.init.kaiming_normal_gain(s)
          self.assertAlmostEqual(
              gain, egain, delta=1e-6, msg=(s, gain, (gain, egain))
          )
  
file_path: objax/tests/nn_init.py
imports: import unittest
  import numpy
  import scipy.stats
  import objax
label_desc: Use image node_id 8 for calling the TestNNInit obj's underlying member method code with example usage: obj.test_kaiming_normal_gain() without return types