torch_diagflat {torch} | R Documentation |
Diagflat
Description
Diagflat
Usage
torch_diagflat(self, offset = 0L)
Arguments
self |
(Tensor) the input tensor. |
offset |
(int, optional) the diagonal to consider. Default: 0 (main diagonal). |
diagflat(input, offset=0) -> Tensor
If
input
is a vector (1-D tensor), then returns a 2-D square tensor with the elements ofinput
as the diagonal.If
input
is a tensor with more than one dimension, then returns a 2-D tensor with diagonal elements equal to a flattenedinput
.
The argument offset
controls which diagonal to consider:
If
offset
= 0, it is the main diagonal.If
offset
> 0, it is above the main diagonal.If
offset
< 0, it is below the main diagonal.
Examples
if (torch_is_installed()) {
a = torch_randn(c(3))
a
torch_diagflat(a)
torch_diagflat(a, 1)
a = torch_randn(c(2, 2))
a
torch_diagflat(a)
}
[Package torch version 0.13.0 Index]