class_docstr: Initializer that generates the identity matrix.
  
  Constructs a 2D identity matrix or batches of these.
class_docstr_tok: ["Initializer","that","generates","the","identity","matrix",".","Constructs","a","2D","identity","matrix","or","batches","of","these","."]
docstr: Constructs an identity initializer.
  
  Args:
    gain: Multiplicative factor to apply to the identity matrix.
docstr_tok: ["Constructs","an","identity","initializer",".","Args",":","gain",":","Multiplicative","factor","to","apply","to","the","identity","matrix","."]
code: def __init__(self, gain: float = 1.0):
      """Constructs an identity initializer.
  
      Args:
        gain: Multiplicative factor to apply to the identity matrix.
      """
      self.gain = gain
  
imports: import abc
  import collections
  from typing import Iterable, Mapping, Optional, Union
  import numpy
  from sonnet.src import types
  import tensorflow
label_desc: Use image node_id 1 to create a new Identity object from inherited base classes: Initializer with example: obj = Identity(gain)