class_docstr: Represents a full axis aligned unit basis.
class_docstr_tok: ["Represents","a","full","axis","aligned","unit","basis","."]
docstr: Initialize a basis of axis aligned unit vectors.
  
  Args:
    event_shape: object representing the shape of the ambient space;
      convertible to `tf.TensorShape`.
    dtype: `Dtype` of this basis.
docstr_tok: ["Initialize","a","basis","of","axis","aligned","unit","vectors",".","Args",":","event_shape",":","object","representing","the","shape","of","the","ambient","space",";","convertible","to","`","tf.TensorShape","`",".","dtype",":","`","Dtype","`","of","this","basis","."]
code: def __init__(self, event_shape, dtype):
      """Initialize a basis of axis aligned unit vectors.
  
      Args:
        event_shape: object representing the shape of the ambient space;
          convertible to `tf.TensorShape`.
        dtype: `Dtype` of this basis.
      """
      self._event_shape = event_shape
      self._dtype = dtype
  
imports: import tensorflow.compat.v2
  from tensorflow_probability.python.internal import distribution_util
  from tensorflow_probability.python.internal import nest_util
  from tensorflow_probability.python.internal import prefer_static
  from tensorflow_probability.python.internal import tensor_util
  from tensorflow_probability.python.internal import tensorshape_util
label_desc: Use image node_id 1 to create a new FullUnitBasis object from inherited base classes: Basis with example: obj = FullUnitBasis(event_shape, dtype)