class_docstr: Tests for linear_operator_unitary.LinearOperatorUnitary.
code: def test_shape(self):
      x = tf.random.stateless_normal(
          [5, 13, 13],
          seed=test_util.test_seed(sampler_type="stateless"),
      )
      q, _ = tf.linalg.qr(x)
      operator = linear_operator_unitary.LinearOperatorUnitary(q)
      self.assertEqual([5], operator.batch_shape)
  
file_path: probability/tensorflow_probability/python/experimental/linalg/linear_operator_unitary_test.py
imports: import tensorflow.compat.v2
  from tensorflow_probability.python.experimental.linalg import linear_operator_unitary
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 2 for calling the LinearOperatorUnitaryTest obj's underlying member method code with example usage: obj.test_shape() without return types