X = np.array([[[1, 22, 4, 78, 2],