Issue with running prediction with multinomial softmax

[ricardoV94]

You need to specify shape manually if you want it to automatically resize with the shape of its arguments. Look at the first example in the docs: pymc.set_data — PyMC 4.3.0+0.gea721e4a.dirty documentation