torch.bmm 批量矩阵相乘 与矩阵相乘torch.matmul()

https://blog.csdn.net/foneone/article/details/103876519

 

官方手册API

https://pytorch.org/docs/stable/torch.html#torch.bmm

你可能感兴趣的:(pytorch)