class_docstr: y_i = x_i
class_docstr_tok: ["y_i","=","x_i"]
code: def A(self) -> tf.Tensor:  # type: ignore[override]
      if self.input_dim is None:
          raise ValueError(
              "An input_dim needs to be specified when using the "
              "`Identity` mean function in combination with expectations."
          )
      return tf.eye(self.input_dim, dtype=default_float())
  
imports: from typing import Collection, Optional, Sequence, Tuple
  import numpy
  import tensorflow
  from check_shapes import check_shape
  from check_shapes import check_shapes, inherit_check_shapes
  from .base import Module, Parameter, TensorType
  from .config import default_float, default_int
label_desc: Use image node_id 3 for calling the Identity obj's underlying member method code with example usage: obj.A() and returns: tf