jaunatisblue/final-qibotn
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse Source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot]
2024-02-08 09:18:18 +00:00
committed by yangliwei
parent b432be6d9c
commit 898ca877e9
10 changed files with 93 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/qibotn/mps_contraction_helper.py
View File

@@ -1,11 +1,10 @@
from cuquantum import CircuitToEinsum, contract, contract_path, tensor
from cuquantum import contract, contract_path
# Reference: https://github.com/NVIDIA/cuQuantum/blob/main/python/samples/cutensornet/tn_algorithms/mps_algorithms.ipynb
class MPSContractionHelper:
"""
A helper class to compute various quantities for a given MPS.
"""A helper class to compute various quantities for a given MPS.
Interleaved format is used to construct the input args for `cuquantum.contract`.
A concrete example on how the modes are populated for a 7-site MPS is provided below:
@@ -43,8 +42,8 @@ class MPSContractionHelper:
]
def contract_norm(self, mps_tensors, options=None):
"""
Contract the corresponding tensor network to form the norm of the MPS.
"""Contract the corresponding tensor network to form the norm of the
MPS.
Args:
mps_tensors: A list of rank-3 ndarray-like tensor objects.
@@ -64,8 +63,8 @@ class MPSContractionHelper:
return self._contract(interleaved_inputs, options=options).real
def contract_state_vector(self, mps_tensors, options=None):
"""
Contract the corresponding tensor network to form the state vector representation of the MPS.
"""Contract the corresponding tensor network to form the state vector
representation of the MPS.
Args:
mps_tensors: A list of rank-3 ndarray-like tensor objects.
@@ -86,8 +85,8 @@ class MPSContractionHelper:
def contract_expectation(
self, mps_tensors, operator, qubits, options=None, normalize=False
):
"""
Contract the corresponding tensor network to form the expectation of the MPS.
"""Contract the corresponding tensor network to form the expectation of
the MPS.
Args:
mps_tensors: A list of rank-3 ndarray-like tensor objects.