class_docstr: Tests correctness of the Y = g(X) = X / norm(X) transformation.
class_docstr_tok: ["Tests","correctness","of","the","Y","=","g","(","X",")","=","X","\/","norm","(","X",")","transformation","."]
code: def testShapeGetters(self):
      x = tf.TensorShape([4])
      y = tf.TensorShape([5])
      uv = unit_vector.UnitVector(validate_args=True)
      self.assertAllEqual(y, uv.forward_event_shape(x))
      self.assertAllEqual(
          tensorshape_util.as_list(y),
          self.evaluate(
              uv.forward_event_shape_tensor(tensorshape_util.as_list(x))
          ),
      )
      self.assertAllEqual(x, uv.inverse_event_shape(y))
      self.assertAllEqual(
          tensorshape_util.as_list(x),
          self.evaluate(
              uv.inverse_event_shape_tensor(tensorshape_util.as_list(y))
          ),
      )
  
file_path: probability/tensorflow_probability/python/bijectors/unit_vector_test.py
imports: import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import bijector_test_util
  from tensorflow_probability.python.bijectors import unit_vector
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 3 for calling the UnitVectorBijectorTest obj's underlying member method code with example usage: obj.testShapeGetters() without return types