Skip to contents

Trace

Source: R/gen-namespace-docs.R, R/gen-namespace-examples.R, R/gen-namespace.R
torch_trace.Rd

Trace

Usage

torch_trace(self)

Arguments

self

the input tensor

trace(input) -> Tensor

Returns the sum of the elements of the diagonal of the input 2-D matrix.

Examples

if (torch_is_installed()) {

x <- torch_arange(1, 9)$view(c(3, 3))
x
torch_trace(x)
}
#> torch_tensor
#> 15
#> [ CPUFloatType{} ]