Expand the test of torch.addbmm and torch.baddbmm (#47079)
Summary:
This is to satisfy the request at https://github.com/pytorch/pytorch/pull/42553#issuecomment-673673914. See also https://github.com/pytorch/pytorch/pull/47124
Pull Request resolved: https://github.com/pytorch/pytorch/pull/47079
Reviewed By: ejguan
Differential Revision: D24735356
Pulled By: ngimel
fbshipit-source-id: 122fceb4902658f350c2fd6f92455adadd0ec2a4