Expand description
Batching module contains several utilities for dealing with batches, such as shuffling and sorting batches
Functions
Filters lists by a max length and returns the ones under the max
Pad all sequences to the length of the longest sequence
Create a pad mask based on the values in the batch (batch shape: batch size, seq len)
Shuffles multiple lists of the same length in the same ways
Sort lists by length. Uses the lengths of the elements in the first list passed in