def test_exp_matrix_rowwise():
inpt = T.matrix()
norm = exp(inpt, axis=1)
f = theano.function([inpt], norm, mode='FAST_COMPILE')
res = f(test_matrix)
correct = np.allclose(res, [114.68499678, 11.47521737])
assert correct, 'exp norm rowwise not working'
def test_exp_matrix_colwise():
inpt = T.matrix()
norm = exp(inpt, axis=0)
f = theano.function([inpt], norm, mode='FAST_COMPILE')
res = f(test_matrix)
correct = np.allclose(res, [2.85361711, 24.90040964, 97.4061874, 1.])
assert correct, 'exp norm colwise not working'
def test_exp_matrix():
inpt = T.matrix()
norm = exp(inpt)
f = theano.function([inpt], norm, mode='FAST_COMPILE')
assert np.allclose(f(test_matrix), 126.16021414942891), 'exp norm for matrix not working'
def test_exp_vector():
inpt = T.vector()
norm = exp(inpt)
f = theano.function([inpt], norm, mode='FAST_COMPILE')
assert np.allclose(f(test_arr), 11.475217368561138), 'exp norm for vector not working'
