pymc.distributions.shape_utils.change_dist_size#
- pymc.distributions.shape_utils.change_dist_size(dist, new_size, expand=False)[source]#
Change or expand the size of a Distribution.
- Parameters
- Returns
A
new
distribution
variable
that
is
equivalent
to
the
original
distribution
with
the
new
size.The
new
distribution
will
not
reuse
the
old
RandomState/Generatorinput
,so
it
will
be
independent
from
the
original
distribution.
Examples
x = Normal.dist(shape=(2, 3)) new_x = change_dist_size(x, new_size=(5, 3), expand=False) assert new_x.eval().shape == (5, 3) new_x = change_dist_size(x, new_size=(5, 3), expand=True) assert new_x.eval().shape == (5, 3, 2, 3)