code: def _get_tensor_shape(self, tensor):
      if self.use_static_shape:
          # If input shapes are static, result shapes should be too.
          return tensorshape_util.as_list(tensor.shape)
      else:
          return self.evaluate(tf.shape(tensor))
  
imports: import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.sts import decomposition
  from tensorflow_probability.python.sts.components import local_linear_trend
  from tensorflow_probability.python.sts.components import seasonal
  from tensorflow_probability.python.sts.components import sum
  from tensorflow_probability.python.sts.forecast import forecast
  from tensorflow_probability.python.sts.internal import missing_values_util
label_desc: Use image node_id 6 for calling the _DecompositionTest obj's underlying member method code with example usage: obj._get_tensor_shape(tensor) and returns: tensorshape_util, self