create_index#

pymc_marketing.mmm.utils.create_index(dims, take)[source]#

Create an index to take the first dimension of a tensor based on the provided dimensions.