pytorch 运行时出现CUBLAS_STATUS_INVALID_VALUE when calling `cublasSgemm`报错

RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)

如题，发现是网络层连接输入与输出维度不匹配导致报错，在使用cpu进行训练下能给出准确的报错信息。