参考 https://blog.youkuaiyun.com/leo_95/article/details/89946318 mm只能是矩阵乘法,即分别为(n×m)和(m×q)bmm是三维张量相乘,即(b×n×m)和(b×m×q),b为批量matmul为张量乘法,任意维度