mm,bmm和matmul的區別

參考 https://blog.csdn.net/leo_95/article/details/89946318

• mm只能是矩陣乘法，即分別爲（n×m）和（m×q）
• bmm是三維張量相乘，即（b×n×m）和（b×m×q），b爲批量
• matmul爲張量乘法，任意維度