Refactor the lora load function for clarity and simplicity

[ajtejankar]

This should prevent some nasty illegal memory access errors

1. Consolidate individual list comprehensions into a single for loop
2. Distinct code to create the lora weight pointers tensor
3. Add test cases to test segments, indices, and weight pointers
4. Add comments explaining the code in some tricky places