mars.tensor.triu¶
- mars.tensor.triu(m, k=0, gpu=None)[源代码]¶
Upper triangle of a tensor.
Return a copy of a matrix with the elements below the k-th diagonal zeroed.
Please refer to the documentation for tril for further details.
参见
tril
lower triangle of a tensor
实际案例
>>> import mars.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]])