I have the following for loop: for j in range(len(list_list_int)): arr_1_, arr_2_, arr_3_ = foo(bar, list_of_ints[j]) arr_1[j,:] = arr_1_.data.numpy() arr_2[j,:] = arr_2_.data.numpy() arr_3[j,...