emu-mps 2.0.1__py3-none-any.whl → 2.0.3__py3-none-any.whl

emu_mps/optimatrix/permutations.py

@@ -1,91 +1,146 @@
-import numpy as np
+import torch
 
 
-def permute_list(input_list: list, permutation: list[int]) -> list:
+def eye_permutation(n: int) -> torch.Tensor:
     """
-    Permutes the input list according to the given permutation.
+    Returns toch.tensor([0, 1, 2, .., n-1])
+    """
+    return torch.arange(n)
+
 
+def permute_list(input_list: list, perm: torch.Tensor) -> list:
+    """
+    Permutes the input list according to the given permutation.
     Parameters
     -------
     input_list :
         A list to permute.
     permutation :
         A list of indices representing the new order.
-
     Returns
     -------
         The permuted list.
-
     Example
     -------
-    >>> permute_list(['a', 'b', 'c'], [2, 0, 1])
+    >>> permute_list(['a','b','c'], torch.tensor([2, 0, 1]))
     ['c', 'a', 'b']
     """
-
-    permuted_list = [None] * len(input_list)
-    for i, p in enumerate(permutation):
-        permuted_list[i] = input_list[p]
-    return permuted_list
+    return [input_list[i] for i in perm.tolist()]
 
 
-def invert_permutation(permutation: list[int]) -> list[int]:
+def permute_tuple(input_tuple: tuple, perm: torch.Tensor) -> tuple:
     """
-    invert_permutation(permutation) -> inv_permutation
+    Permutes the input tuple according to the given permutation.
+    Parameters
+    -------
+    input_tuple :
+        A tuple to permute.
+    permutation :
+        A tuple of indices representing the new order.
+    Returns
+    -------
+        The permuted tuple.
+    Example
+    -------
+    >>> permute_tuple(('a','b','c'), torch.tensor([2, 0, 1]))
+    ('c', 'a', 'b')
+    """
+    lst_elem = list(input_tuple)
+    return tuple(permute_list(lst_elem, perm))
 
-    Inverts the input permutation list.
 
+def permute_string(input_str: str, perm: torch.Tensor) -> str:
+    """
+    Permutes the input string according to the given permutation.
     Parameters
     -------
+    input_string :
+        A string to permute.
     permutation :
-        A list of indices representing the order
-
+        A list of indices representing the new order.
     Returns
     -------
-        permutation list inverse to the input list
-
-    Example:
+        The permuted string.
+    Example
     -------
-    >>> invert_permutation([2, 0, 1])
-    [1, 2, 0]
+    >>> permute_string("abc", torch.tensor([2, 0, 1]))
+    'cab'
     """
-
-    inv_perm = np.empty_like(permutation)
-    inv_perm[permutation] = np.arange(len(permutation))
-    return list(inv_perm)
+    permuted = permute_list(list(input_str), perm)
+    return "".join(permuted)
 
 
-def permute_matrix(mat: np.ndarray, permutation: list[int]) -> np.ndarray:
+def inv_permutation(permutation: torch.Tensor) -> torch.Tensor:
     """
-    permute_matrix(matrix, permutation_list) -> permuted_matrix
+    inv_permutation(permutation) -> inverted_perm
 
-    Simultaneously permutes columns and rows according to a permutation list.
+    Inverts the input permutation list.
 
     Parameters
     -------
-    matrix :
-        square matrix nxn
     permutation :
-        permutation list
+        A list of indices representing the order
 
     Returns
     -------
-        matrix with permuted columns and rows
+        permutation list inverse to the input list
 
     Example:
     -------
-    >>> matrix = np.array([
-    ...    [1, 2, 3],
-    ...    [4, 5, 6],
-    ...    [7, 8, 9]])
-    >>> permutation = [1, 0, 2]
-    >>> permute_matrix(matrix, permutation)
-    array([[5, 4, 6],
-           [2, 1, 3],
-           [8, 7, 9]])
+    >>> inv_permutation(torch.tensor([2, 0, 1]))
+    tensor([1, 2, 0])
     """
+    inv_perm = torch.empty_like(permutation)
+    inv_perm[permutation] = torch.arange(len(permutation))
+    return inv_perm
+
 
-    perm = np.array(permutation)
-    return mat[perm, :][:, perm]
+def permute_tensor(tensor: torch.Tensor, perm: torch.Tensor) -> torch.Tensor:
+    """
+    Permute a 1D or square 2D torch tensor using the given permutation indices.
+    For 1D tensors, applies the permutation to the elements.
+    For 2D square tensors, applies the same permutation to both rows and columns.
+
+    Parameters
+    ----------
+    tensor : torch.Tensor
+        A 1D or 2D square tensor to be permuted.
+    perm : torch.Tensor
+        A 1D tensor of indices specifying the permutation order.
+
+    Returns
+    -------
+    torch.Tensor
+        A new tensor with elements (1D) or rows and columns (2D) permuted according to `perm`.
+
+    Raises
+    ------
+    ValueError
+        If tensor is not 1D or square 2D.
+
+    Examples
+    --------
+    >>> vector = torch.tensor([10, 20, 30])
+    >>> perm = torch.tensor([2, 0, 1])
+    >>> permute_tensor(vector, perm)
+    tensor([30, 10, 20])
+
+    >>> matrix = torch.tensor([
+    ...     [1, 2, 3],
+    ...     [4, 5, 6],
+    ...     [7, 8, 9]])
+    >>> perm = torch.tensor([1, 0, 2])
+    >>> permute_tensor(matrix, perm)
+    tensor([[5, 4, 6],
+            [2, 1, 3],
+            [8, 7, 9]])
+    """
+    if tensor.ndim == 1:
+        return tensor[perm]
+    elif tensor.ndim == 2 and tensor.shape[0] == tensor.shape[1]:
+        return tensor[perm][:, perm]
+    else:
+        raise ValueError("Only 1D tensors or square 2D tensors are supported.")
 
 
 if __name__ == "__main__":

emu_mps/tdvp.py

@@ -73,12 +73,20 @@ def apply_effective_Hamiltonian(
     # this order seems to be pretty balanced, but needs to be
     # revisited when use-cases are more well-known
     state = torch.tensordot(left_bath, state, 1)
-    state = torch.tensordot(state, ham, ([1, 2], [0, 2]))
-    state = torch.tensordot(state, right_bath, ([3, 1], [1, 2]))
+    state = state.permute(0, 3, 1, 2)
+    ham = ham.permute(0, 2, 1, 3)
+    state = state.reshape(state.shape[0], state.shape[1], -1).contiguous()
+    ham = ham.reshape(-1, ham.shape[2], ham.shape[3]).contiguous()
+    state = torch.tensordot(state, ham, 1)
+    state = state.permute(0, 2, 1, 3)
+    state = state.reshape(state.shape[0], state.shape[1], -1).contiguous()
+    right_bath = right_bath.permute(2, 1, 0)
+    right_bath = right_bath.reshape(-1, right_bath.shape[2]).contiguous()
+    state = torch.tensordot(state, right_bath, 1)
     return state
 
 
-_TIME_CONVERSION_COEFF = 0.001  # Omega and delta are given in rad/ms, dt in ns
+_TIME_CONVERSION_COEFF = 0.001  # Omega and delta are given in rad/μs, dt in ns
 
 
 def evolve_pair(

{emu_mps-2.0.1.dist-info → emu_mps-2.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: emu-mps
-Version: 2.0.1
+Version: 2.0.3
 Summary: Pasqal MPS based pulse emulator built on PyTorch
 Project-URL: Documentation, https://pasqal-io.github.io/emulators/
 Project-URL: Repository, https://github.com/pasqal-io/emulators
@@ -25,7 +25,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
 Requires-Python: >=3.10
-Requires-Dist: emu-base==2.0.1
+Requires-Dist: emu-base==2.0.3
 Description-Content-Type: text/markdown
 
 <div align="center">

emu_mps-2.0.3.dist-info/RECORD

@@ -0,0 +1,19 @@
+emu_mps/__init__.py,sha256=wGG0u1wHUcq_MuDjygyApshJY_ome-TZMVggIfHcgns,734
+emu_mps/algebra.py,sha256=ngPtTH-j2ZCBWoaJZXlkUyIlug7dY7Q92gzfnRlpPMA,5485
+emu_mps/custom_callback_implementations.py,sha256=CUs0kW3HRaPE7UeFNQOFbeWJMsz4hS2q4rgS57BBp-A,2411
+emu_mps/hamiltonian.py,sha256=gOPxNOBmk6jRPPjevERuCP_scGv0EKYeAJ0uxooihes,15622
+emu_mps/mpo.py,sha256=1ogQ25GZCwMzZ_m449oGHcYyDKrofBCr1eyzzrIPMhQ,8824
+emu_mps/mps.py,sha256=GIiWxctNmHARgf-PgQc6IHKNCe5HYSnbtlXI6Hc-0wI,20085
+emu_mps/mps_backend.py,sha256=bS83qFxvdoK-c12_1WaPw6O7xUc7vdWifZNHUzNP5sM,2091
+emu_mps/mps_backend_impl.py,sha256=8aPSn6ZGFf4DtTodzajX6WJtQHI9F8eMUN1AbrPzbgw,25611
+emu_mps/mps_config.py,sha256=JCK_frUVyMRJjI0NPU-dpCg-uO3JKeVzQMOLDkQ9XVk,8104
+emu_mps/noise.py,sha256=h4X2EFjoC_Ok0gZ8I9wN77RANXaVehTBbjkcbY_GAmY,784
+emu_mps/observables.py,sha256=7GQDH5kyaVNrwckk2f8ZJRV9Ca4jKhWWDsOCqYWsoEk,1349
+emu_mps/tdvp.py,sha256=0qTw9qhg0WbaAyBgeTpULHrNL0ytj80ZUb1P6GKD7Ww,6172
+emu_mps/utils.py,sha256=BqRJYAcXqprtZVJ0V_j954ON2bhTdtZiaTojsYyrWrg,8193
+emu_mps/optimatrix/__init__.py,sha256=fBXQ7-rgDro4hcaBijCGhx3J69W96qcw5_3mWc7tND4,364
+emu_mps/optimatrix/optimiser.py,sha256=k9suYmKLKlaZ7ozFuIqvXHyCBoCtGgkX1mpen9GOdOo,6977
+emu_mps/optimatrix/permutations.py,sha256=9DDMZtrGGZ01b9F3GkzHR3paX4qNtZiPoI7Z_Kia3Lc,3727
+emu_mps-2.0.3.dist-info/METADATA,sha256=ZU1USoiZ12em9lo4GgAWFiGE0oBT83yXitpm2fx8LvA,3505
+emu_mps-2.0.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+emu_mps-2.0.3.dist-info/RECORD,,

emu_mps-2.0.1.dist-info/RECORD

@@ -1,19 +0,0 @@
-emu_mps/__init__.py,sha256=GIR0FJOvmMHby7Vx_Da3w7YqqLpOxa0E8E3poxXsapY,734
-emu_mps/algebra.py,sha256=ngPtTH-j2ZCBWoaJZXlkUyIlug7dY7Q92gzfnRlpPMA,5485
-emu_mps/custom_callback_implementations.py,sha256=CUs0kW3HRaPE7UeFNQOFbeWJMsz4hS2q4rgS57BBp-A,2411
-emu_mps/hamiltonian.py,sha256=LcBs6CKBb643a1e9AAVtQoUfa4L_0dIhLOKecx5OOWs,15864
-emu_mps/mpo.py,sha256=wSonS6i3zEt3yRTgyZ7F6vT51pUKavFcLOxFFphBv8k,8793
-emu_mps/mps.py,sha256=KqAjo-nxgM-xQSg1NFNchwXKoPRcrKuuycFMsWr7iX8,19610
-emu_mps/mps_backend.py,sha256=_3rlg6XeI4fHaDiJRfPL6pDkX9k48hAHKXd8fkvkOFs,2004
-emu_mps/mps_backend_impl.py,sha256=M_do7QRBLAoHfwF_EpyMCb6g7w7BSs5hsPa5UE0z9Zs,22958
-emu_mps/mps_config.py,sha256=89nu5OhNUX31eAeeYvvKnAHegpPVD43jH5Nmp635HVU,6984
-emu_mps/noise.py,sha256=h4X2EFjoC_Ok0gZ8I9wN77RANXaVehTBbjkcbY_GAmY,784
-emu_mps/observables.py,sha256=7GQDH5kyaVNrwckk2f8ZJRV9Ca4jKhWWDsOCqYWsoEk,1349
-emu_mps/tdvp.py,sha256=pIQ2NXA2Mrkp3elhqQbX3pdJVbtKkG3c5r9fFlJo7pI,5755
-emu_mps/utils.py,sha256=BqRJYAcXqprtZVJ0V_j954ON2bhTdtZiaTojsYyrWrg,8193
-emu_mps/optimatrix/__init__.py,sha256=lHWYNeThHp57ZrwTwXd0p8bNvcCv0w_AZ31iCWflBUo,226
-emu_mps/optimatrix/optimiser.py,sha256=7j9_jMQC-Uh2DzdIVB44InRzZO6AbbGhvmm7lC6N3tk,6737
-emu_mps/optimatrix/permutations.py,sha256=JRXGont8B4QgbkV9CzrA0w7uzLgBrmZ1J9aa0G52hPo,1979
-emu_mps-2.0.1.dist-info/METADATA,sha256=aGt2cFa9rplEaJAo8z4J5uoAkqq9WnQIlOhciUc6vkg,3505
-emu_mps-2.0.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-emu_mps-2.0.1.dist-info/RECORD,,