Add shadows.py and repair the typo in gates.py #169
Conversation
local synchronization
local synchronization
|
please add type annotation, test files |
tensorcircuit/shadows.py
Outdated
| c_.R(i, theta=theta, alpha=alpha, phi=phi) | ||
| return c_.sample(batch=repeat, format="sample_bin") | ||
|
|
||
| vpm = backend.vmap(proj_measure, vectorized_argnums=0) |
tensorcircuit/shadows.py
Outdated
|
|
||
| v = backend.vmap(tensor_prod, vectorized_argnums=0) | ||
| vv = backend.vmap(v, vectorized_argnums=0) | ||
| return vv(shadow_states) |
There was a problem hiding this comment.
is there a final sum lacking?
tensorcircuit/shadows.py
Outdated
| new_indices = f"{ABC[0:2 * nq:2]}{ABC[1:2 * nq:2]}" | ||
|
|
||
| def tensor_prod(dms): | ||
| return backend.reshape(backend.einsum(f'{",".join(old_indices)}->{new_indices}', *dms, optimize=True), |
There was a problem hiding this comment.
not sure whether every backend has optimizer=True argument, to be checked
tests/test_shadows.py
Outdated
| expc, ent, lss_states, sdw_state = csjit(c.state(), pauli_strings, status, sub, x, y, z) | ||
| expc = np.median(expc) | ||
|
|
||
| assert np.abs(expc - exact) < 0.1 |
tests/test_shadows.py
Outdated
| def classical_shadow(psi, pauli_strings, status, sub, x, y, z): | ||
| snapshots = shadow_snapshots(psi, pauli_strings, status) | ||
| lss_states = local_snapshot_states(snapshots, pauli_strings) | ||
| sdw_state = global_shadow_state(lss_states, sub=sub) |
There was a problem hiding this comment.
could also add a test to assert close between global state from local states and snapshots directly
tests/test_shadows.py
Outdated
| sdw_state = global_shadow_state(lss_states) | ||
|
|
||
| R = np.array(sdw_state - bell_state) | ||
| assert np.sqrt(np.trace(R.conj().T @ R)) < 0.1 |
There was a problem hiding this comment.
np.testing.assert_allclose could directly assert the closeness of two matrix elementwisely
add gitattr to auto lr
Codecov Report
@@ Coverage Diff @@
## master #169 +/- ##
==========================================
- Coverage 75.04% 74.99% -0.06%
==========================================
Files 66 67 +1
Lines 10167 10340 +173
==========================================
+ Hits 7630 7754 +124
- Misses 2537 2586 +49
📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more |
tests/test_shadows.py
Outdated
| lss_states = shadow_snapshots(c.state(), pauli_strings, status) | ||
| sdw_state = global_shadow_state(lss_states) | ||
|
|
||
| np.allclose(sdw_state, bell_state, atol=0.01) |
There was a problem hiding this comment.
np.tesing.assert_allclose()
refraction-ray
merged commit fbe5a9e
into
tencent-quantum-lab:master
|
@all-contributors please add @PeilinZHENG for code, tests, docs |
|
I've put up a pull request to add @PeilinZHENG! 🎉 |
No description provided.