Fix Bag-input

[rs9000]

Hi,
Need a fix to handle models which require multiple inputs.
My fix is very ungly, but the idea is iterate on input list and add mini-batch only if it required.

[FrancescoSaverioZuppichini]

Correct me if I am wrong, if the inputs is a tensor with shape >2 then it won't have the batch dim in the first. So you want to be able to pass multiple inputs to your model at once, am I right?

[rs9000]

I want to pass as input an array which contains tensors that already have the mini batch size.

Many models require two inputs, vqa for example, require image and question.
They are usually packed into an array where input[0] is the image and input[1] is the question.

My fix want iterate over the input array and for each tensor we add mini-batch only if the shape is equal to 1. Because tensors with shape > 1 I suppose already have a mini-batch size.