I have 2 lists l1, l2 that contain 5 numpy arrays each of the following shapes:
l1
l2
[(32, 84, 84, 1), (32,), (32,), (32,), (32, 84, 84,