fix batching for no. of inputs < batch size