Description
输入两个整数类型的矩阵mat1(m行d列)和mat2(d行n列),返回矩阵相乘后的结果mat1*mat2(m行n列)。矩阵均用二维列表进行表示。
Input
两个矩阵分别严格满足m d 和 d n的形状(m>=1, d>=1, n>=1,具体数值需要根据输入确定),矩阵中的元素均为整数。
Output
返回相乘后的矩阵,用二维列表表示,每一个元素均为整数
其余要求同首题。
Sample Input
[[1,2]]
[[1],[2]]
Sample Output
[[5]]
Reference Code
def main(mat1,mat2):return [[sum(mat1[i][s]*mat2[s][j] for s in range(len(mat1[0]))) for j in range(len(mat2[0]))] for i in range(len(mat1))]
print(main(eval(input()),eval(input())))