def check_two_tensor_operation(function_name, x_input_shape, y_input_shape,
**kwargs):
xval = np.random.random(x_input_shape) - 0.5
xth = BTH.variable(xval)
xnp = BNP.variable(xval)
yval = np.random.random(y_input_shape) - 0.5
yth = BTH.variable(yval)
ynp = BNP.variable(yval)
_zth = getattr(BTH, function_name)(xth, yth, **kwargs)
zth = BTH.eval(_zth)
znp = BNP.eval(getattr(BNP, function_name)(xnp, ynp, **kwargs))
assert zth.shape == znp.shape
assert_allclose(zth, znp, atol=1e-05)
def test_tile():
shape = (3, 4)
arr = np.arange(np.prod(shape)).reshape(shape)
arr_th = BTH.variable(arr)
arr_np = BNP.variable(arr)
n = (2, 1)
th_z = BTH.tile(arr_th, n)
th_rep = BTH.eval(th_z)
np_rep = BNP.eval(BNP.tile(arr_np, n))
assert_allclose(np_rep, th_rep, atol=1e-05)
def check_single_tensor_operation(function_name, input_shape, **kwargs):
val = np.random.random(input_shape) - 0.5
xth = BTH.variable(val)
xnp = BNP.variable(val)
_zth = getattr(BTH, function_name)(xth, **kwargs)
zth = BTH.eval(_zth)
znp = BNP.eval(getattr(BNP, function_name)(xnp, **kwargs))
assert zth.shape == znp.shape
assert_allclose(zth, znp, atol=1e-05)
def test_switch():
val = np.random.random()
xth = BTH.variable(val)
xth = BTH.ifelse(xth >= 0.5, xth * 0.1, xth * 0.2)
xnp = BNP.variable(val)
xnp = BNP.ifelse(xnp >= 0.5, xnp * 0.1, xnp * 0.2)
zth = BTH.eval(xth)
znp = BNP.eval(xnp)
assert zth.shape == znp.shape
assert_allclose(zth, znp, atol=1e-05)
def test_shape_operations():
# concatenate
xval = np.random.random((4, 3))
xth = BTH.variable(xval)
xnp = BNP.variable(xval)
yval = np.random.random((4, 2))
yth = BTH.variable(yval)
ynp = BNP.variable(yval)
zth = BTH.eval(BTH.concatenate([xth, yth], axis=-1))
znp = BNP.eval(BNP.concatenate([xnp, ynp], axis=-1))
assert zth.shape == znp.shape
assert_allclose(zth, znp, atol=1e-05)
check_single_tensor_operation('reshape', (4, 2), shape=(8, 1))
check_single_tensor_operation('permute_dimensions', (4, 2, 3),
pattern=(2, 0, 1))
check_single_tensor_operation('repeat', (4, 1), n=3)
check_single_tensor_operation('flatten', (4, 1))
check_single_tensor_operation('squeeze', (4, 3, 1), axis=2)
check_single_tensor_operation('squeeze', (4, 1, 1), axis=1)
check_composed_tensor_operations('reshape', {'shape': (4, 3, 1, 1)},
'squeeze', {'axis': 2}, (4, 3, 1, 1))
def test_value_manipulation():
val = np.random.random((4, 2))
xth = BTH.variable(val)
xnp = BNP.variable(val)
# get_value
valth = BTH.get_value(xth)
valnp = BNP.get_value(xnp)
assert valnp.shape == valth.shape
assert_allclose(valth, valnp, atol=1e-05)
# set_value
BTH.set_value(xth, val)
valth = BTH.get_value(xth)
assert valnp.shape == val.shape
assert_allclose(valth, val, atol=1e-05)
def test_repeat_elements():
reps = 3
for ndims in [1, 2, 3]:
shape = np.arange(2, 2 + ndims)
arr = np.arange(np.prod(shape)).reshape(shape)
arr_th = BTH.variable(arr)
arr_np = BNP.variable(arr)
for rep_axis in range(ndims):
np_rep = np.repeat(arr, reps, axis=rep_axis)
th_z = BTH.repeat_elements(arr_th, reps, axis=rep_axis)
th_rep = BTH.eval(th_z)
bnp_rep = BNP.eval(BNP.repeat_elements(arr_np, reps,
axis=rep_axis))
assert th_rep.shape == np_rep.shape
assert bnp_rep.shape == np_rep.shape
assert_allclose(np_rep, th_rep, atol=1e-05)
assert_allclose(np_rep, bnp_rep, atol=1e-05)
def test_svd():
input_shape = (4, 4)
val = np.random.random(input_shape) - 0.5
xth = BTH.variable(val)
xnp = BNP.variable(val)
Uth, Sth, Vth = BTH.svd(xth)
Unp, Snp, Vnp = BNP.svd(xnp)
Uth = Uth.eval()
Sth = Sth.eval()
Vth = Vth.eval()
assert Uth.shape == Unp.shape
assert Sth.shape == Snp.shape
assert Vth.shape == Vnp.shape
assert_allclose(Uth, Unp, atol=1e-05)
assert_allclose(Sth, Snp, atol=1e-05)
assert_allclose(Vth, Vnp, atol=1e-05)
def check_composed_tensor_operations(first_function_name, first_function_args,
second_function_name,
second_function_args, input_shape):
''' Creates a random tensor t0 with shape input_shape and compute
t1 = first_function_name(t0, **first_function_args)
t2 = second_function_name(t1, **second_function_args)
with both Theano and TensorFlow backends and ensures the answers match.
'''
val = np.random.random(input_shape) - 0.5
xth = BTH.variable(val)
xnp = BNP.variable(val)
yth = getattr(BTH, first_function_name)(xth, **first_function_args)
ynp = getattr(BNP, first_function_name)(xnp, **first_function_args)
zth = BTH.eval(
getattr(BTH, second_function_name)(yth, **second_function_args))
znp = BNP.eval(
getattr(BNP, second_function_name)(ynp, **second_function_args))
assert zth.shape == znp.shape
assert_allclose(zth, znp, atol=1e-05)