#NUMPY067. 矩阵乘积的对角线

矩阵乘积的对角线

描述

给定两个 n×nn \times n 矩阵,计算它们的乘积矩阵的对角线元素。

格式

输入

第一行是整数 nn(矩阵大小,1n101 \le n \le 10

接下来 2n2n 行,每 nn 行一个矩阵,每行 nn 个整数。

输出

打印乘积矩阵的对角线元素(数组形式)。

样例

2
1 2
3 4
5 6
7 8
[23 34]

提示

np.dot\texttt{np.dot} —— 矩阵乘法;np.diag\texttt{np.diag} —— 获取对角线