class_docstr: Marginal distribution of a Multitask GP at finitely many points.
docstr: Helper to `sample`; sets static shape info.
docstr_tok: ["Helper","to","`","sample","`",";","sets","static","shape","info","."]
code: def _set_sample_static_shape(
      self, x, sample_shape, index_points=None
  ):
      """Helper to `sample`; sets static shape info."""
      batch_shape = self._batch_shape(index_points=index_points)
      event_shape = tf.TensorShape(
          self._event_shape(index_points=index_points)
      )
      return distribution._set_sample_static_shape_for_tensor(  # pylint:disable=protected-access
          x,
          sample_shape=sample_shape,
          event_shape=event_shape,
          batch_shape=batch_shape,
      )
  
imports: import functools
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import cholesky_util
  from tensorflow_probability.python.distributions import distribution
  from tensorflow_probability.python.distributions import mvn_linear_operator
  from tensorflow_probability.python.distributions.internal import stochastic_process_util
  from tensorflow_probability.python.experimental.linalg import linear_operator_unitary
  from tensorflow_probability.python.experimental.psd_kernels import multitask_kernel
  from tensorflow_probability.python.internal import batch_shape_lib
  from tensorflow_probability.python.internal import distribution_util
  from tensorflow_probability.python.internal import dtype_util
  from tensorflow_probability.python.internal import nest_util
  from tensorflow_probability.python.internal import parameter_properties
  from tensorflow_probability.python.internal import prefer_static
  from tensorflow_probability.python.internal import reparameterization
  from tensorflow_probability.python.internal import tensor_util
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow.python.util import nest
label_desc: Use image node_id 19 for calling the MultiTaskGaussianProcess obj's underlying member method code with example usage: obj._set_sample_static_shape(x, sample_shape, index_points) and returns: distribution