def test_complex(self):
rng = np.random.default_rng(2333)
m = fmatrix()
c = at_complex(m[0], m[1])
assert c.type == cvector
r, i = [real(c), imag(c)]
assert r.type == fvector
assert i.type == fvector
f = aesara.function([m], [r, i])
mval = np.asarray(rng.standard_normal((2, 5)), dtype="float32")
rval, ival = f(mval)
assert np.all(rval == mval[0]), (rval, mval[0])
assert np.all(ival == mval[1]), (ival, mval[1])
def f(a, b):
ac = at_complex(a[0], a[1])
return abs((ac * b)**2).sum()
def f(a):
ac = at_complex(a[0], a[1])
return abs(ac).sum()
def f(m):
c = at_complex(m[0], m[1])
return 0.5 * real(c) + 0.9 * imag(c)
def f(m):
c = at_complex(m[0], m[1])
return 0.5 * abs(c)