Triu

## triu(input, diagonal=0, out=NULL) -> Tensor

Returns the upper triangular part of a matrix (2-D tensor) or batch of matrices
`input`

, the other elements of the result tensor `out`

are set to 0.

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

The argument `diagonal`

controls which diagonal to consider. If
`diagonal`

= 0, all elements on and above the main diagonal are
retained. A positive value excludes just as many diagonals above the main
diagonal, and similarly a negative value includes just as many diagonals below
the main diagonal. The main diagonal are the set of indices
\(\lbrace (i, i) \rbrace\) for \(i \in [0, \min\{d_{1}, d_{2}\} - 1]\) where
\(d_{1}, d_{2}\) are the dimensions of the matrix.

## Examples

```
if (torch_is_installed()) {
a = torch_randn(c(3, 3))
a
torch_triu(a)
torch_triu(a, diagonal=1)
torch_triu(a, diagonal=-1)
b = torch_randn(c(4, 6))
b
torch_triu(b, diagonal=1)
torch_triu(b, diagonal=-1)
}
#> torch_tensor
#> 2.4724 -0.6489 -0.4529 0.9390 0.7321 0.4581
#> -1.4422 -1.2979 0.3688 0.7509 1.2400 1.9312
#> 0.0000 0.3239 0.5506 0.1849 -0.4237 -1.5429
#> 0.0000 0.0000 -0.3917 -0.2252 0.9236 -1.2134
#> [ CPUFloatType{4,6} ]
```