mars.tensor.tril¶
- mars.tensor.tril(m, k=0, gpu=None)[源代码]¶
Lower triangle of a tensor.
Return a copy of a tensor with elements above the k-th diagonal zeroed.
- 参数
- 返回
tril – Lower triangle of m, of same shape and data-type as m.
- 返回类型
Tensor, shape (M, N)
参见
triu
same thing, only for the upper triangle
实际案例
>>> import mars.tensor as mt
>>> mt.tril([[1,2,3],[4,5,6],[7,8,9],[10,11,12]], -1).execute() array([[ 0, 0, 0], [ 4, 0, 0], [ 7, 8, 0], [10, 11, 12]])