def get_steps(x, shape):
    steps = shape[1]
    if x is None:
        return [None for step in range(steps)]
    xs = x.reshape(shape + (-1,))
    return [xs[:, step, :] for step in range(steps)]
Convert a (feature_size, steps * batch_size) array
dataset/ETHPy150Open NervanaSystems/neon/neon/layers/recurrent.py/get_steps