bmm uses baddbmm under the hood: https://github.com/pytorch/pytorch/blob/master/torch/lib/ATen/Declarations.cwrap#L3189-L3212 7. Moreover, variable names are just names… Once you pass them to another function, they lose that info. So trying to align variable names is not really useful.
@jytug 's answer is correct.