torch_diag: Diag

In torch: Tensors and Neural Networks with 'GPU' Acceleration

Diag

Description

Diag

Usage

torch_diag(self, diagonal = 0L)

Arguments

self	(Tensor) the input tensor.
diagonal	(int, optional) the diagonal to consider

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

• If input is a vector (1-D tensor), then returns a 2-D square tensor with the elements of input as the diagonal.

• If input is a matrix (2-D tensor), then returns a 1-D tensor with the diagonal elements of input.

The argument diagonal controls which diagonal to consider:

• If diagonal = 0, it is the main diagonal.

• If diagonal > 0, it is above the main diagonal.

• If diagonal < 0, it is below the main diagonal.

torch documentation built on May 29, 2024, 9:54 a.m.