Exemple #1
0
 def batch(self, dataset, batch_size):
     batch_fn_map = {"features": shifted_batch}
     return diverse_batch(
         dataset,
         batch_size,
         batch_fn_map,
         default_batch_fn=lambda dataset, batch_size: dataset.padded_batch(
             batch_size, get_padded_shapes(dataset)))
Exemple #2
0
 def batch(self, dataset, batch_size):
     return dataset.padded_batch(batch_size, get_padded_shapes(dataset))