maxframe.tensor.triu

maxframe.tensor.triu(m, k=0, gpu=None)

返回张量的上三角部分。

返回一个矩阵的副本，其中第 k 条对角线以下的元素被置为零。

更多细节请参阅 tril 的文档。

参见

tril

返回张量的下三角部分

示例

>>> import maxframe.tensor as mt

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