我希望在 Python 中定義的函數(shù)中找到矩陣的蹤跡（使用 Numpy）。輸入?yún)?shù)tensor和tensor_transpose都是大小為 (N,2,2) 的矩陣，并且是從 VTK 文件中提取的（N 是一個相當(dāng)大的數(shù)字，并且因文件而異）。所以A和B都是 (N,2,2) 的數(shù)組。通過獲取每個數(shù)組的軌跡（對角項的總和），應(yīng)該返回每個數(shù)組的單個值。所以np.trace(A)**3)-(np.trace(B)**3應(yīng)該是一個單一的數(shù)值，數(shù)組的形狀為 (N,1)。我的輸出雖然沒有顯示這一點，但返回的形狀是(2,).誰能解釋為什么？這是trace功能的問題嗎？有解決方案嗎？import numpy as npA=np.array(0.5*(tensor-tensor_transpose))B=np.array(0.5*(tensor+tensor_transpose))C=np.array(0.5*((np.trace(A)**3)-(np.trace(B)**3)))print(A.shape)print(B.shape)print(C.shape)#Output#(60600, 2, 2)#(60600, 2, 2)#(2,)