index_variable#
- pymc_marketing.mmm.components.base.index_variable(var, dims, idx)[source]#
Index a variable based on the provided dimensions and index.
- Parameters:
- var
TensorVariable The variable to index.
- dims
tuple[str, …] The dims of the variable.
- idx
dict[str,pt.TensorLike] The index to use for the variable.
- var
- Returns:
TensorVariableThe indexed variable.