Hello Aman! Yes, I was able to somewhat get it working using the suggestions by @KFrank  here: Multiply a model by trainable scalar 18
It does work, I just didn’t automate the process to generate the mirror paths. Might or might not be useful in your case, but I hope at least it gives you an idea