def test_l1_matrix_colwise():
inpt = T.matrix()
norm = l1(inpt, axis=0)
f = theano.function([inpt], norm, mode='FAST_COMPILE')
res = f(test_matrix)
correct = np.allclose(res, [3., 4.2, 6.5, 100.2])
assert correct, 'l1 norm colwise not working'
def test_l1_matrix_rowwise():
inpt = T.matrix()
norm = l1(inpt, axis=1)
f = theano.function([inpt], norm, mode='FAST_COMPILE')
res = f(test_matrix)
correct = np.allclose(res, [109.9, 4.])
assert correct, 'l1 norm rowwise not working'
def test_l1_matrix():
inpt = T.matrix()
norm = l1(inpt)
f = theano.function([inpt], norm, mode='FAST_COMPILE')
assert np.allclose(f(test_matrix), 113.9), 'l1 norm for matrix not working'
def test_l1_vector():
inpt = T.vector()
norm = l1(inpt)
f = theano.function([inpt], norm, mode='FAST_COMPILE')
assert f(test_arr) == 4, 'l1 norm for vector not working'
