zero or more batch dimensions. The batch dimensions of
        "LU_pivots" must be equal to the batch dimensions of
        "LU_data".
Keyword Arguments:
      out (Tensor, optional) -- the output tensor.
Example:
  >>> A = torch.randn(2, 3, 3)
  >>> b = torch.randn(2, 3, 1)
  >>> LU, pivots = torch.linalg.lu_factor(A)
  >>> x = torch.lu_solve(b, LU, pivots)
  >>> torch.dist(A @ x, b)
  tensor(1.00000e-07 *
         2.8312)
