-
Notifications
You must be signed in to change notification settings - Fork 583
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add
tfq.math.mps_1d_expectation for 1D MPS (#610)
* initial commit * Fix simulate_mps_test * Add CheckQubitsIn1D * Add Eigen & QSim MPS * Add mps_1d op and its test * Mike's feedback - add bond_dim as attr * Add Attr and its tests * Rename to mps_1d_expectation * Fix lint and format * Add import_test * Fix wheel test - add mps into release/BUILD * Mike's feedback * WIP where Segmentation fault happens? * Add Mike's feedback - no gate with control_qubit is allowed * Revert control_qubit gate validation. Both Operation with natural born controlled gate (e.g. CNOT) and synthesized ControlledOperation are converted into the same tfq_gate_set with control_qubit. So, we can't filter ControlledOperation in this way, revert the previous commit. * Fix program_resolution_test * Fix CheckQubitsIn1D() * Add print messages * Add debug print * Bump up to qsim==0.10.2 * Remove tfq::QsimFor and util_qsim.h::ComputeExpectationMPS * Add ComputeExpectationMPSQsim in util_qsim.h * Add more detailed debug prints * Bump up bond_dim from 2 (default) to 4 to fix segfault error. * Fix how to use ApplyGate in ComputeExpectationMPSQsim * Uncomment ComputeSmall() and Remove debug outputs * Fix format * Mike's feedback 1 : use fuses circuit and ApplyFusedGate() * Mike's feedback 2 : set default bond_dim to 4 * Fix format * Mike's feedback 3 : bump up to qsim==0.10.3 * bring 1d mps expectation into working state. * empty test. Co-authored-by: Michael Broughton <[email protected]>
- Loading branch information
1 parent
768ac15
commit dea26c6
Showing
14 changed files
with
905 additions
and
15 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,57 @@ | ||
| # Copyright 2020 The TensorFlow Quantum Authors. All Rights Reserved. | ||
| # | ||
| # Licensed under the Apache License, Version 2.0 (the "License"); | ||
| # you may not use this file except in compliance with the License. | ||
| # You may obtain a copy of the License at | ||
| # | ||
| # http://www.apache.org/licenses/LICENSE-2.0 | ||
| # | ||
| # Unless required by applicable law or agreed to in writing, software | ||
| # distributed under the License is distributed on an "AS IS" BASIS, | ||
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| # See the License for the specific language governing permissions and | ||
| # limitations under the License. | ||
| # ============================================================================== | ||
| """Module to register MPS simulation ops.""" | ||
| import os | ||
| import tensorflow as tf | ||
| from tensorflow_quantum.core.ops.load_module import load_module | ||
|
|
||
| MATH_OP_MODULE = load_module(os.path.join("math_ops", "_tfq_math_ops.so")) | ||
|
|
||
|
|
||
| def mps_1d_expectation(programs, | ||
| symbol_names, | ||
| symbol_values, | ||
| pauli_sums, | ||
| bond_dim=4): | ||
| """Calculate the expectation value of circuits wrt some operator(s) | ||
| Args: | ||
| programs: `tf.Tensor` of strings with shape [batch_size] containing | ||
| the string representations of the circuits to be executed. | ||
| symbol_names: `tf.Tensor` of strings with shape [n_params], which | ||
| is used to specify the order in which the values in | ||
| `symbol_values` should be placed inside of the circuits in | ||
| `programs`. | ||
| symbol_values: `tf.Tensor` of real numbers with shape | ||
| [batch_size, n_params] specifying parameter values to resolve | ||
| into the circuits specificed by programs, following the ordering | ||
| dictated by `symbol_names`. | ||
| pauli_sums: `tf.Tensor` of strings with shape [batch_size, n_ops] | ||
| containing the string representation of the operators that will | ||
| be used on all of the circuits in the expectation calculations. | ||
| bond_dim: Integer value used for the bond dimension during simulation. | ||
| Returns: | ||
| `tf.Tensor` with shape [batch_size, n_ops] that holds the | ||
| expectation value for each circuit with each op applied to it | ||
| (after resolving the corresponding parameters in). | ||
| """ | ||
| return MATH_OP_MODULE.tfq_simulate_mps1d_expectation(programs, | ||
| symbol_names, | ||
| tf.cast( | ||
| symbol_values, | ||
| tf.float32), | ||
| pauli_sums, | ||
| bond_dim=bond_dim) |
Oops, something went wrong.