vaeac_extend_batch {shapr} | R Documentation |
Extends Incomplete Batches by Sampling Extra Data from Dataloader
Description
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.
Usage
vaeac_extend_batch(batch, dataloader, batch_size)
Arguments
batch |
The batch we want to check if has the right size, and if not extend it until it has the right size. |
dataloader |
A |
batch_size |
Integer. The number of samples to include in each batch. |
Value
Returns the extended batch with the correct batch_size.
Author(s)
Lars Henry Berge Olsen
[Package shapr version 1.0.4 Index]