modify reshape layer
parallelize