torch (version 0.0.2)

torch_trace: Trace

Description

Trace

trace(input) -> Tensor

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

Examples

# NOT RUN {
if (torch_is_installed()) {

x = torch_arange(1., 10.)$view(c(3, 3))
x
torch_trace(x)
}
# }