pytensor.dot works exactly like numpy.dot. You can call X.shape.eval() and β[cluster_index,:].shape.eval() to see what shapes you are getting and it should be easy to see why it’s not working.
1 Like