Python dask.array.trace用法及代碼示例

用法: dask.array.trace(a, offset=0, axis1=0, axis2=1, dtype=None)

返回沿數組對角線的總和。

此文檔字符串是從 numpy.trace 複製的。

可能存在與 Dask 版本的一些不一致之處。

如果a 是二維的，則返回沿其對角線與給定偏移量的總和，即所有 i 的元素 a[i,i+offset] 的總和。

如果a 有兩個以上的維度，則由axis1 和axis2 指定的軸用於確定返回其軌跡的二維子數組。結果數組的形狀與 a 的形狀相同，但刪除了 axis1 和 axis2。

a：array_like: 輸入數組，從中獲取對角線。
offset：整數，可選: 對角線與主對角線的偏移量。可以是正麵的也可以是負麵的。默認為 0。
axis1, axis2：整數，可選: 用作對角線的二維子陣列的第一和第二軸的軸。默認值是 a 的前兩個軸。
dtype：dtype，可選: 確定返回數組和元素相加的累加器的數據類型。如果 dtype 的值為 None 並且 a 是精度小於默認整數精度的整數類型，則使用默認整數精度。否則，精度與 a 相同。
out：ndarray，可選(在 Dask 中不支持): 放置輸出的數組。它的類型被保留，它必須具有正確的形狀來保存輸出。

sum_along_diagonals：ndarray: 如果a 是二維的，則返回沿對角線的和。如果a 具有更大的維度，則返回沿對角線的和數組。

>>> np.trace(np.eye(3))  
3.0
>>> a = np.arange(8).reshape((2,2,2))  
>>> np.trace(a)  
array([6, 8])

>>> a = np.arange(24).reshape((2,2,2,3))  
>>> np.trace(a).shape  
(2, 3)

相關用法

注：本文由純淨天空篩選整理自dask.org大神的英文原創作品 dask.array.trace。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。