get_symbolic_rv_shape#

pymc_marketing.model_graph.get_symbolic_rv_shape(rv, raise_if_rv_in_graph=True)[source]#

Get the symbolic shape of the random variables in the graph.

Parameters:
rvVariable

The random variable to get the shape of.

raise_if_rv_in_graphbool, optional

Whether to raise an error if the random variable is still in the graph.

Returns:
tuple[TensorVariable]

The shape of the random variables.