No "RuntimeError: mat1 and mat2 shapes cannot be multiplied" when using GPU

You might be hitting this known and already fixed issue so please update to the latest release (1.9.1).