tensorcircuit-nightly 1.3.0.dev20250809__py3-none-any.whl → 1.3.0.dev20250810__py3-none-any.whl

tests/test_torchnn.py

@@ -1,99 +0,0 @@
-import os
-import sys
-import numpy as np
-import pytest
-from pytest_lazyfixture import lazy_fixture as lf
-
-thisfile = os.path.abspath(__file__)
-modulepath = os.path.dirname(os.path.dirname(thisfile))
-
-sys.path.insert(0, modulepath)
-
-import tensorcircuit as tc
-
-try:
-    import torch
-except ImportError:
-    pytest.skip("torch is not installed", allow_module_level=True)
-
-
-@pytest.mark.parametrize("backend", [lf("tfb"), lf("jaxb"), lf("torchb")])
-def test_quantumnet(backend):
-    n = 6
-    nlayers = 2
-
-    def qpred(x, weights):
-        c = tc.Circuit(n)
-        for i in range(n):
-            c.rx(i, theta=x[i])
-        for j in range(nlayers):
-            for i in range(n - 1):
-                c.cnot(i, i + 1)
-            for i in range(n):
-                c.rx(i, theta=weights[2 * j, i])
-                c.ry(i, theta=weights[2 * j + 1, i])
-        ypred = tc.backend.stack([c.expectation_ps(x=[i]) for i in range(n)])
-        return tc.backend.real(ypred)
-
-    if tc.backend.name == "pytorch":
-        use_interface = False
-    else:
-        use_interface = True
-
-    ql = tc.TorchLayer(
-        qpred, weights_shape=[2 * nlayers, n], use_interface=use_interface
-    )
-
-    yp = ql(torch.ones([3, n]))
-    print(yp)
-
-    np.testing.assert_allclose(yp.shape, np.array([3, n]))
-
-
-@pytest.mark.parametrize("backend", [lf("tfb"), lf("jaxb")])
-def test_inputs_multiple(backend):
-    n = 3
-    p = 0.1
-    K = tc.backend
-    torchb = tc.get_backend("pytorch")
-
-    def f(state, noise, weights):
-        c = tc.Circuit(n, inputs=state)
-        for i in range(n):
-            c.rz(i, theta=weights[i])
-        for i in range(n):
-            c.depolarizing(i, px=p, py=p, pz=p, status=noise[i])
-        return K.real(c.expectation_ps(x=[0]))
-
-    layer = tc.TorchLayer(f, [n], use_vmap=True, vectorized_argnums=[0, 1])
-    state = torchb.ones([2, 2**n]) / 2 ** (n / 2)
-    noise = 0.2 * torchb.ones([2, n], dtype="float32")
-    l = layer(state, noise)
-    lsum = torchb.sum(l)
-    print(l)
-    lsum.backward()
-    for p in layer.parameters():
-        print(p.grad)
-
-
-@pytest.mark.parametrize("backend", [lf("tfb"), lf("jaxb"), lf("torchb")])
-def test_torchnn_hardware(backend):
-    n = 2
-
-    def qf(inputs, param):
-        inputs = tc.backend.convert_to_tensor(tc.get_backend("pytorch").numpy(inputs))
-        param = tc.backend.convert_to_tensor(tc.get_backend("pytorch").numpy(param))
-
-        c = tc.Circuit(n)
-        c.rx(0, theta=inputs[0])
-        c.rx(1, theta=inputs[1])
-        c.h(1)
-        c.rzz(0, 1, theta=param[0])
-        r = tc.backend.stack([c.expectation_ps(z=[i]) for i in range(n)])
-
-        r = tc.get_backend("pytorch").convert_to_tensor(tc.backend.numpy(r))
-        return torch.real(r)
-
-    ql = tc.torchnn.HardwareNet(qf, [1])
-    qnet = torch.nn.Sequential(ql, torch.nn.Linear(2, 1))
-    print(qnet(torch.ones([5, 2])))

tests/test_van.py

@@ -1,102 +0,0 @@
-import sys
-import os
-import itertools
-
-thisfile = os.path.abspath(__file__)
-modulepath = os.path.dirname(os.path.dirname(thisfile))
-
-sys.path.insert(0, modulepath)
-
-import numpy as np
-import tensorflow as tf
-
-
-from tensorcircuit.applications.van import MaskedLinear, ResidualBlock, MADE, PixelCNN
-
-
-def test_masklinear():
-    ml = MaskedLinear(5, 10, 3, mask=tf.zeros([10, 3, 5, 3]))
-    ml.set_weights([tf.ones([10, 3, 5, 3]), tf.ones([10, 3])])
-    tf.debugging.assert_near(ml(tf.ones([1, 5, 3])), tf.ones([1, 10, 3]))
-
-    mask = np.zeros([10, 3, 5, 3])
-    mask[3, 2, 1, 0] = 1.0
-    ml = MaskedLinear(5, 10, 3, mask=tf.constant(mask, dtype=tf.float32))
-    ml.set_weights([tf.ones([10, 3, 5, 3]), tf.zeros([10, 3])])
-    assert tf.reduce_sum(ml(tf.ones([5, 3]))[3, :]) == 1.0
-
-    w = tf.random.uniform(shape=[10, 3, 5, 3])
-    b = tf.random.uniform(shape=[10, 3])
-    w_m = tf.reshape(w, [30, 15])
-    b_m = tf.reshape(b, [30, 1])
-    inputs = tf.ones([5, 3])
-    inputs_m = tf.reshape(inputs, [15, 1])
-    r_m = w_m @ inputs_m + b_m
-    r = tf.reshape(r_m, [10, 3])
-    ml = MaskedLinear(5, 10, 3)
-    ml.set_weights([w, b])
-    tf.debugging.assert_near(ml(inputs), r)
-
-
-def test_residual_block():
-    dense1 = tf.keras.layers.Dense(10, use_bias=False)
-    dense1.build([1, 1])
-    dense1.set_weights([np.ones([1, 10])])
-    dense2 = tf.keras.layers.Dense(1, use_bias=False)
-    dense2.build([1, 10])
-    dense2.set_weights([np.ones([10, 1])])
-    m = ResidualBlock([dense1, dense2])
-    assert m(tf.ones([1, 1])) == 11.0
-
-
-def test_made():
-    import itertools
-
-    m = MADE(2, 2, 6, 3, 3, nonmerge=False)
-    l = []
-    for i in itertools.product(*[list(range(3)) for _ in range(2)]):
-        l.append(list(i))
-    basis = tf.constant(l, dtype=tf.int32)
-    print(basis)
-    ptot = tf.reduce_sum(tf.exp(m.log_prob(tf.one_hot(basis, depth=3))))
-    np.testing.assert_allclose(ptot.numpy(), 1.0, atol=1e-5)
-
-    s, logp = m.sample(10)
-    print(logp)
-    assert s.shape == (10, 2, 3)
-
-
-def test_made_fit_peak():
-    opt = tf.optimizers.Adam(learning_rate=0.01)
-    m = MADE(5, 5, 4, 3, 2)
-    for step in range(100):
-        with tf.GradientTape() as t:
-            loss = -tf.reduce_sum(
-                m.log_prob(
-                    tf.one_hot(
-                        [[0, 0, 0, 0, 0], [1, 1, 1, 1, 1], [2, 2, 1, 1, 0]], depth=3
-                    )
-                )
-            )
-            gr = t.gradient(loss, m.variables)
-        if step % 20 == 0:
-            print(
-                tf.exp(
-                    m.log_prob(
-                        tf.one_hot(
-                            [[0, 0, 0, 0, 0], [1, 1, 1, 1, 1], [2, 2, 1, 1, 0]], depth=3
-                        )
-                    )
-                ).numpy()
-            )
-        opt.apply_gradients(zip(gr, m.variables))
-
-
-def test_pixelcnn():
-    m = PixelCNN(3, 5, 8)
-    l = []
-    for i in itertools.product(*[list(range(3)) for _ in range(4)]):
-        l.append(list(i))
-    basis = tf.constant(tf.reshape(l, [-1, 2, 2]), dtype=tf.int32)
-    ptot = tf.reduce_sum(tf.exp(m.log_prob(tf.one_hot(basis, depth=3))))
-    np.testing.assert_allclose(ptot.numpy(), 1.0, atol=1e-5)