class_docstr: Applies a Bijector to the diagonal of a matrix.
  
  #### Example
  
  ```python
  b = tfb.TransformDiagonal(diag_bijector=tfb.Exp())
  
  b.forward([[1., 0.],
             [0., 1.]])
  # ==> [[2.718, 0.],
         [0., 2.718]]
  ```
class_docstr_tok: ["Applies","a","Bijector","to","the","diagonal","of","a","matrix",".","#","#","#","#","Example","``","`","python","b","=","tfb.TransformDiagonal","(","diag_bijector=tfb.Exp","(",")",")","b.forward","(","[","[","1.",",","0",".","]",",","[","0.",",","1",".","]","]",")","#","==",">","[","[","2.718",",","0",".","]",",","[","0.",",","2.718","]","]","``","`"]
code: def diag_bijector(self):
      return self._diag_bijector
  
file_path: probability/tensorflow_probability/python/bijectors/transform_diagonal.py
imports: import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import bijector
label_desc: Use image node_id 3 for calling the TransformDiagonal obj's underlying member method code with example usage: obj.diag_bijector() and returns: self