torch: torch_lu – R documentation

Become an expert in R — Interactive courses, Cheat Sheets, certificates and more!

torch_lu

Description

Computes the LU factorization of a matrix or batches of matrices A. Returns a tuple containing the LU factorization and pivots of A. Pivoting is done if pivot is set to True.

Usage

torch_lu(A, pivot = TRUE, get_infos = FALSE, out = NULL)

Arguments

`A`	(Tensor) the tensor to factor of size (, m, n)(,m,n)
`pivot`	(bool, optional) – controls whether pivoting is done. Default: TRUE
`get_infos`	(bool, optional) – if set to True, returns an info IntTensor. Default: FALSE
`out`	(tuple, optional) – optional output tuple. If get_infos is True, then the elements in the tuple are Tensor, IntTensor, and IntTensor. If get_infos is False, then the elements in the tuple are Tensor, IntTensor. Default: NULL

Examples

if (torch_is_installed()) {

A = torch_randn(c(2, 3, 3))
torch_lu(A)

}

torch

Tensors and Neural Networks with 'GPU' Acceleration

v0.3.0

MIT + file LICENSE

Authors

Daniel Falbel [aut, cre, cph], Javier Luraschi [aut], Dmitriy Selivanov [ctb], Athos Damiani [ctb], Christophe Regouby [ctb], Krzysztof Joachimiak [ctb], RStudio [cph]

Initial release