Exemplos de get_padded_shapes em Python

Exemplo n.º 1

 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)))

Exemplo n.º 2

 def batch(self, dataset, batch_size):
     return dataset.padded_batch(batch_size, get_padded_shapes(dataset))