(self)
| 2621 | tensor_shape.TensorShape([batch_size]).concatenate(self._dataset_shape)) |
| 2622 | |
| 2623 | def _unbatch(self): |
| 2624 | if self._dataset_shape.ndims == 0: |
| 2625 | raise ValueError("Unbatching a dataset is only supported for rank >= 1") |
| 2626 | return DatasetSpec(self._element_spec, self._dataset_shape[1:]) |
| 2627 | |
| 2628 | def _to_batched_tensor_list(self, value): |
| 2629 | if self._dataset_shape.ndims == 0: |