Python numpy tril用法及代碼示例

本文簡要介紹 python 語言中 numpy.tril 的用法。

用法:

numpy.tril(m, k=0)

數組的下三角形。

返回一個數組的副本，其元素高於k-th 對角線歸零。對於具有ndim超過 2,tril將適用於最後兩個軸。

參數：

m： 數組，形狀(…，M，N)

輸入數組。

k： 整數，可選

對角線高於其零元素。 k = 0(默認)是主對角線，k < 0 在其下方，k > 0 在上方。

返回：

tril： ndarray，形狀(...，M，N)

m 的下三角形，與 m 具有相同的形狀和數據類型。

例子：

>>> np.tril([[1,2,3],[4,5,6],[7,8,9],[10,11,12]], -1)
array([[ 0,  0,  0],
       [ 4,  0,  0],
       [ 7,  8,  0],
       [10, 11, 12]])

>>> np.tril(np.arange(3*4*5).reshape(3, 4, 5))
array([[[ 0,  0,  0,  0,  0],
        [ 5,  6,  0,  0,  0],
        [10, 11, 12,  0,  0],
        [15, 16, 17, 18,  0]],
       [[20,  0,  0,  0,  0],
        [25, 26,  0,  0,  0],
        [30, 31, 32,  0,  0],
        [35, 36, 37, 38,  0]],
       [[40,  0,  0,  0,  0],
        [45, 46,  0,  0,  0],
        [50, 51, 52,  0,  0],
        [55, 56, 57, 58,  0]]])