Python ComplexTensor.numpy Examples

Example #1

def test_inv(ch):
    torch.manual_seed(100)
    X = ComplexTensor(torch.rand(2, 3, ch, ch), torch.rand(2, 3, ch, ch))
    X = X + X.conj().transpose(-1, -2)
    assert FC.allclose(ComplexTensor(np.linalg.inv(X.numpy())),
                       inv(X),
                       atol=1e-4)

Example #2

File: test_functional.py Project: mtxing/pytorch_complex

def test_trace():
    t = ComplexTensor(_get_complex_array(10, 10))
    x = numpy.trace(t.numpy())
    y = F.trace(t).numpy()
    numpy.testing.assert_allclose(x, y)