mars.tensor.tril#

mars.tensor.tril(m, k=0, gpu=None)[source]#

Lower triangle of a tensor.

Return a copy of a tensor with elements above the k-th diagonal zeroed.

Parameters

  • m (array_like, shape (M, N)) – Input tensor.

  • k (int, optional) – Diagonal above which to zero elements. k = 0 (the default) is the main diagonal, k < 0 is below it and k > 0 is above.

  • gpu (bool, optional) – Allocate the tensor on GPU if True, None as default

Returns

tril – Lower triangle of m, of same shape and data-type as m.

Return type

Tensor, shape (M, N)

See also

triu

same thing, only for the upper triangle

Examples

>>> 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]])