dataX = np.expand_dims(np.random.randint(0,10,size=100), axis = 0) dataY = np.expand_dims(np.random.randint(0,10,size=100), axis = 0)