[ 0.2634, -0.9255, -0.3459]]])
      >>> pivots
      tensor([[ 3,  3,  3],
              [ 3,  3,  3]], dtype=torch.int32)
      >>> A_LU, pivots, info = torch.lu(A, get_infos=True)
      >>> if info.nonzero().size(0) == 0:
      ...     print('LU factorization succeeded for all samples!')
      LU factorization succeeded for all samples!