pytorch --- torch.bmm()

谢和颂
2023-12-01

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

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

 类似资料: