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

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot]
2024-03-04 07:02:09 +00:00
parent f295bfd276
commit cda817faf9
2 changed files with 4 additions and 5 deletions

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