numpy向量的转置与向量相乘

• ai是一个向量(np列表)
• 直接a1.T@a1返回的是标量

    for ai in a:
        res += ai.T@ai /(1-ai.dot(x))       
    res += np.diag(2(1+x**2)/((1-x**2).dot(1-x**2)))

为了返回$a_i.T@a_i$，可以把给ai扩充一个第一维度，转为矩阵
另一种方法是从a中选取 子矩阵

    for i in range(len(a)):
        res += a[i:i+1].T@a[i:i+1] /(1-a[i].dot(x))