If the height of the batch is less than batch_size, this function extends the batch with
data from the torch::dataloader() until the batch reaches the required size.
Note that batch is a tensor.
vaeac_extend_batch(batch, dataloader, batch_size)Returns the extended batch with the correct batch_size.
The batch we want to check if has the right size, and if not extend it until it has the right size.
A torch::dataloader() object from which we can create an iterator object
and load data to extend the batch.
Integer. The number of samples to include in each batch.
Lars Henry Berge Olsen