Add TensorVariable & remove broadcast_dist_samples_shape

Author: UditNayak

docs/source/api/shape_utils.rst

@@ -16,4 +16,3 @@ This module introduces functions that are made aware of the requested `size_tupl
    to_tuple
    rv_size_is_none
    change_dist_size
-   broadcast_dist_samples_shape

pymc/distributions/shape_utils.py

@@ -339,7 +339,7 @@ def change_dist_size(
     ----------
     dist : TensorVariable
         The old distribution to be resized.
-    new_size : Union[int, Tuple[int, ...]]
+    new_size : Union[int, Tuple[int, ...], TensorVariable]
         The new size of the distribution.
     expand : bool, optional
         If True, `new_size` is prepended to the existing distribution `size`,