description: Truncates a list of segments using a pre-determined truncation strategy.
Truncates a list of segments using a pre-determined truncation strategy.
@abc.abstractmethodgenerate_mask( segments )
Generates a boolean mask specifying which portions of segments to drop.
Users should be able to use the results of generate_mask() to drop items in
segments using tf.ragged.boolean_mask(seg, mask).
| Args | |
|---|---|
| `segments` | A list of `RaggedTensor` each w/ a shape of [num_batch, (num_items)]. |
| Returns | |
|---|---|
| a list with len(segments) number of items and where each item is a `RaggedTensor` with the same shape as its counterpart in `segments` and with a boolean dtype where each value is True if the corresponding value in `segments` should be kept and False if it should be dropped instead. |
trim(
segments
)
Truncate the list of segments.
Truncate the list of segments using the truncation strategy defined by
generate_mask.
| Args | |
|---|---|
| `segments` | A list of `RaggedTensor`s w/ shape [num_batch, (num_items)]. |