I got the correct result with torch1.2, while the result from torch1.8 is not reasonable. If there are something wrong with torch.bmm()?
Could you update to the latest stable release and check, if you would still be seeing unexpected results?
