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

Merge branch 'doc-fix' of https://github.com/qiboteam/qibotn into doc-fix

Browse Source
This commit is contained in:
tankya2
2024-03-04 15:49:09 +08:00
parent ac9fa2106e cda817faf9
commit 07f433f24e
2 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/qibotn/mps_contraction_helper.py
View File

@@ -7,7 +7,7 @@ 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`. Interleaved format is used to construct the input args for `cuquantum.contract`.
Reference: https://github.com/NVIDIA/cuQuantum/blob/main/python/samples/cutensornet/tn_algorithms/mps_algorithms.ipynb Reference: https://github.com/NVIDIA/cuQuantum/blob/main/python/samples/cutensornet/tn_algorithms/mps_algorithms.ipynb
The following compute quantities are supported: The following compute quantities are supported:

7
src/qibotn/mps_utils.py
View File

@@ -3,9 +3,8 @@ from cuquantum import contract
from cuquantum.cutensornet.experimental import contract_decompose from cuquantum.cutensornet.experimental import contract_decompose
def initial(num_qubits, dtype): def initial(num_qubits, dtype):
"""Generate the MPS with an initial state of :math:`\ket{00...00}`""" r"""Generate the MPS with an initial state of :math:`\ket{00...00}`"""
state_tensor = cp.asarray([1, 0], dtype=dtype).reshape(1, 2, 1) state_tensor = cp.asarray([1, 0], dtype=dtype).reshape(1, 2, 1)
mps_tensors = [state_tensor] * num_qubits mps_tensors = [state_tensor] * num_qubits
return mps_tensors return mps_tensors
@@ -26,9 +25,9 @@ def mps_site_right_swap(mps_tensors, i, **kwargs):
def apply_gate(mps_tensors, gate, qubits, **kwargs): def apply_gate(mps_tensors, gate, qubits, **kwargs):
"""Apply the gate operand to the MPS tensors in-place. """Apply the gate operand to the MPS tensors in-place.
# Reference: https://github.com/NVIDIA/cuQuantum/blob/main/python/samples/cutensornet/tn_algorithms/mps_algorithms.ipynb # Reference: https://github.com/NVIDIA/cuQuantum/blob/main/python/samples/cutensornet/tn_algorithms/mps_algorithms.ipynb
Args: Args:
mps_tensors: A list of rank-3 ndarray-like tensor objects. mps_tensors: A list of rank-3 ndarray-like tensor objects.
The indices of the ith tensor are expected to be the bonding index to the i-1 tensor, The indices of the ith tensor are expected to be the bonding index to the i-1 tensor,