The error seems to be raised in GMMConv and I guess that gaussian might be created in the wrong shape.
I would recommend to add print statements into the message function and check the shape of x_j as well as gaussian and make sure they match for the desired operations.