A `Dataset` that maps a function over elements in its input.
| 3512 | |
| 3513 | |
| 3514 | class MapDataset(UnaryDataset): |
| 3515 | """A `Dataset` that maps a function over elements in its input.""" |
| 3516 | |
| 3517 | def __init__(self, |
| 3518 | input_dataset, |
| 3519 | map_func, |
| 3520 | use_inter_op_parallelism=True, |
| 3521 | preserve_cardinality=False, |
| 3522 | use_legacy_function=False): |
| 3523 | """See `Dataset.map()` for details.""" |
| 3524 | self._input_dataset = input_dataset |
| 3525 | self._use_inter_op_parallelism = use_inter_op_parallelism |
| 3526 | self._preserve_cardinality = preserve_cardinality |
| 3527 | self._map_func = StructuredFunctionWrapper( |
| 3528 | map_func, |
| 3529 | self._transformation_name(), |
| 3530 | dataset=input_dataset, |
| 3531 | use_legacy_function=use_legacy_function) |
| 3532 | variant_tensor = gen_dataset_ops.map_dataset( |
| 3533 | input_dataset._variant_tensor, # pylint: disable=protected-access |
| 3534 | self._map_func.function.captured_inputs, |
| 3535 | f=self._map_func.function, |
| 3536 | use_inter_op_parallelism=self._use_inter_op_parallelism, |
| 3537 | preserve_cardinality=self._preserve_cardinality, |
| 3538 | **self._flat_structure) |
| 3539 | super(MapDataset, self).__init__(input_dataset, variant_tensor) |
| 3540 | |
| 3541 | def _functions(self): |
| 3542 | return [self._map_func] |
| 3543 | |
| 3544 | @property |
| 3545 | def element_spec(self): |
| 3546 | return self._map_func.output_structure |
| 3547 | |
| 3548 | def _transformation_name(self): |
| 3549 | return "Dataset.map()" |
| 3550 | |
| 3551 | |
| 3552 | class ParallelMapDataset(UnaryDataset): |
no outgoing calls
no test coverage detected