docstr: Identity
docstr_tok: ["Identity"]
code: def test_identity(self):
      """Identity"""
      for s in ((16, 16), (24, 10000), (10000, 32)):
          for gain in (1, 2):
              init = np.asarray(objax.nn.init.identity(s, gain=gain))
              diff_sum = np.linalg.norm(
                  init - gain * np.eye(s[0], s[1])
              )
              self.assertAlmostEqual(
                  diff_sum, 0, delta=1e-2, msg=(s, gain)
              )
  
file_path: objax/tests/nn_init.py
imports: import unittest
  import numpy
  import scipy.stats
  import objax
label_desc: Use image node_id 6 for calling the TestNNInit obj's underlying member method code with example usage: obj.test_identity() without return types