https://blog.csdn.net/foneone/article/details/103876519 官方手册API https://pytorch.org/docs/stable/torch.html#torch.bmm