transfer_to_batch_gpu returns null when input has primitives
Describe the bug
when passing a batch such as:
batch = list(tensor, tensor, [0,1,2])  
the list of ints won't be returned correctly
Additional context
Fix should add a return of the item it no condition matches