import numpy as np
import theano.tensor as T
from theano import function

x0 = T.fmatrix()
x1 = T.fmatrix()
y0 = T.concatenate([x0, x1], axis=0)
y1 = T.concatenate([x0, x1], axis=1)

f0 = function(inputs=[x0, x1], outputs=y0)
f1 = function(inputs=[x0, x1], outputs=y1)

a = np.array([[1,2], [3,4]], dtype=np.float32)
b = np.array([[5,6], [7,8]], dtype=np.float32)

z0 = f0(a,b)
z1 = f1(a,b)

print z0
print z1



