关于torch.bmm()函数计算过程

不少框架中提供的矩阵乘法都是出于简化计算的考虑，不少状况下在进行计算时候都会牵扯到 batch size 这一个维度，这就使得不少矩阵的计算是三维的，Pytorch中的bmm()函数就能够很方便的实现三维数组的乘法，而不用拆成二维数组使用for循环解决。在查资料的时候发现有些博客写的有些小地方不太对，并且有不少提问都是关于 bmm（）函数具体是如何计算的，所以记录。 web 文章目录 1.torc

>>阅读原文<<