Using scan to loop across differenet length of vectors

Since the set dimensions are known in advance you can just use python if statements, and then concatenate the learning rates from the 3 scans for whatever you need downstream.