pytorch --- torch.bmm()

小批量乘法运算 torch.bmm()

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