小批量乘法运算 torch.bmm() 假设第⼀个⼩批量中包含n个形状为[a, b]的矩阵 ,第⼆个⼩批量中包含n个形状为[b, c] 的矩阵 。这两个⼩批量的矩阵乘法输出为n个形状为[a, c] 的矩阵 。因此,给定两个形状分别为( n ,a ,b )和(n ,b ,c )的 Tensor ,⼩批量乘法输出的形状为(n ,a ,c )