Use this line to parallelize the loop for in the layers: `#pragma omp parallel for num_threads(/*number of logical cores*/) schedule(runtime)`
Use this line to parallelize the loop for in the layers:
#pragma omp parallel for num_threads(/*number of logical cores*/) schedule(runtime)