triu¶

dragon.triu(
  inputs,
  k=0,
  copy=True,
  **kwargs
)[source]¶

Return the upper triangular part of input.

\[\text{out}_{ij} = \begin{cases} 0, & \text{ if } j < i + k \\ \text{input}_{ij}, & \text{ otherwise } \end{cases} \]

Examples:

x = dragon.ones((3, 3))
print(dragon.triu(x))

Parameters:

• inputs (dragon.Tensor) – The input tensor.
• k (int, optional, default=0) – Diagonal below which to zero elements.
• copy (bool, optional, default=True) – Return a new tensor or call in-place.

Returns:

dragon.Tensor – The output tensor.