Returns default padding tensors in a structure matching `input_dataset`.
(input_dataset)
| 3410 | |
| 3411 | |
| 3412 | def _default_padding(input_dataset): |
| 3413 | """Returns default padding tensors in a structure matching `input_dataset`.""" |
| 3414 | def make_zero(t): |
| 3415 | if t.base_dtype == dtypes.string: |
| 3416 | return "" |
| 3417 | elif t.base_dtype == dtypes.variant: |
| 3418 | error_msg = ("Unable to create padding for field of type 'variant' " |
| 3419 | "because t.base_type == dtypes.variant == " |
| 3420 | "{}.".format( |
| 3421 | t.base_dtype)) |
| 3422 | raise TypeError(error_msg) |
| 3423 | else: |
| 3424 | return np.zeros_like(t.as_numpy_dtype()) |
| 3425 | |
| 3426 | return nest.map_structure( |
| 3427 | make_zero, get_legacy_output_types(input_dataset)) |
| 3428 | |
| 3429 | |
| 3430 | class PaddedBatchDataset(UnaryDataset): |
no test coverage detected