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.
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)
}
#> [[1]]
#> torch_tensor
#> (1,.,.) =
#> 1.1530 -1.0274 -1.3670
#> 0.8862 -0.9511 -0.6307
#> -0.2910 -0.0160 -0.8346
#>
#> (2,.,.) =
#> -3.1277 -0.4164 -0.6328
#> 0.3718 1.7323 -0.6936
#> 0.2817 0.0421 0.0635
#> [ CPUFloatType{2,3,3} ]
#>
#> [[2]]
#> torch_tensor
#> 3 2 3
#> 2 3 3
#> [ CPUIntType{2,3} ]
#>