dpnp.tensor.triu

dpnp.tensor.triu(x, /, *, k=0)

Returns the upper triangular part of a matrix (or a stack of matrices) x.

The upper triangular part of the matrix is defined as the elements on and above the specified diagonal k.

Parameters:

• x (usm_ndarray) -- Input array.

• k (int, optional) --

  Specifies the diagonal below which to set elements to zero. If k = 0, the diagonal is the main diagonal. If k < 0, the diagonal is below the main diagonal. If k > 0, the diagonal is above the main diagonal.

  Default: 0.

Returns:

out -- An upper-triangular array or a stack of upper-triangular arrays.

Return type:

usm_ndarray