A `Dataset` skipping the first `count` elements from its input.
| 3221 | |
| 3222 | |
| 3223 | class SkipDataset(UnaryUnchangedStructureDataset): |
| 3224 | """A `Dataset` skipping the first `count` elements from its input.""" |
| 3225 | |
| 3226 | def __init__(self, input_dataset, count): |
| 3227 | """See `Dataset.skip()` for details.""" |
| 3228 | self._input_dataset = input_dataset |
| 3229 | self._count = ops.convert_to_tensor(count, dtype=dtypes.int64, name="count") |
| 3230 | variant_tensor = gen_dataset_ops.skip_dataset( |
| 3231 | input_dataset._variant_tensor, # pylint: disable=protected-access |
| 3232 | count=self._count, |
| 3233 | **self._flat_structure) |
| 3234 | super(SkipDataset, self).__init__(input_dataset, variant_tensor) |
| 3235 | |
| 3236 | |
| 3237 | class ShardDataset(UnaryUnchangedStructureDataset): |