def __init__(self, input_dataset, window_size, window_shift, window_stride): """See `sliding_window_batch` for details.""" self._input_dataset = input_dataset self._window_size = ops.convert_to_tensor(window_size, dtype=dtypes.int64, name="window_stride") self._window_stride = ops.convert_to_tensor(window_stride, dtype=dtypes.int64, name="window_stride") self._window_shift = ops.convert_to_tensor(window_shift, dtype=dtypes.int64, name="window_shift") input_structure = dataset_ops.get_structure(input_dataset) self._structure = nest.map_structure( lambda component_spec: component_spec._batch(None), input_structure) # pylint: disable=protected-access variant_tensor = ged_ops.experimental_sliding_window_dataset( self._input_dataset._variant_tensor, # pylint: disable=protected-access window_size=self._window_size, window_shift=self._window_shift, window_stride=self._window_stride, **self._flat_structure) super(_SlideDataset, self).__init__(input_dataset, variant_tensor)
def __init__(self, input_dataset, window_size, window_shift, window_stride): """See `sliding_window_batch` for details.""" self._input_dataset = input_dataset self._window_size = ops.convert_to_tensor(window_size, dtype=dtypes.int64, name="window_stride") self._window_stride = ops.convert_to_tensor(window_stride, dtype=dtypes.int64, name="window_stride") self._window_shift = ops.convert_to_tensor(window_shift, dtype=dtypes.int64, name="window_shift") input_structure = dataset_ops.get_structure(input_dataset) self._structure = input_structure._batch(None) # pylint: disable=protected-access if compat.forward_compatible(2019, 8, 3): variant_tensor = ged_ops.sliding_window_dataset( self._input_dataset._variant_tensor, # pylint: disable=protected-access window_size=self._window_size, window_shift=self._window_shift, window_stride=self._window_stride, **self._flat_structure) else: variant_tensor = ged_ops.experimental_sliding_window_dataset( self._input_dataset._variant_tensor, # pylint: disable=protected-access window_size=self._window_size, window_shift=self._window_shift, window_stride=self._window_stride, **self._flat_structure) super(_SlideDataset, self).__init__(input_dataset, variant_tensor)
def _as_variant_tensor(self): return ged_ops.experimental_sliding_window_dataset( self._input_dataset._as_variant_tensor(), # pylint: disable=protected-access window_size=self._window_size, window_shift=self._window_shift, window_stride=self._window_stride, **dataset_ops.flat_structure(self))
def __init__(self, input_dataset, window_size, window_shift, window_stride): """See `sliding_window_batch` for details.""" self._input_dataset = input_dataset self._window_size = ops.convert_to_tensor(window_size, dtype=dtypes.int64, name="window_stride") self._window_stride = ops.convert_to_tensor(window_stride, dtype=dtypes.int64, name="window_stride") self._window_shift = ops.convert_to_tensor(window_shift, dtype=dtypes.int64, name="window_shift") input_structure = structure.convert_legacy_structure( input_dataset.output_types, input_dataset.output_shapes, input_dataset.output_classes) self._structure = input_structure._batch(None) # pylint: disable=protected-access variant_tensor = ged_ops.experimental_sliding_window_dataset( self._input_dataset._variant_tensor, # pylint: disable=protected-access window_size=self._window_size, window_shift=self._window_shift, window_stride=self._window_stride, **dataset_ops.flat_structure(self)) super(_SlideDataset, self).__init__(input_dataset, variant_tensor)
def _as_variant_tensor(self): return ged_ops.experimental_sliding_window_dataset( self._input_dataset._as_variant_tensor(), # pylint: disable=protected-access window_size=self._window_size, window_shift=self._window_shift, window_stride=self._window_stride, **dataset_ops.flat_structure(structure=self._output_structure))
def __init__(self, input_dataset, window_size, window_shift, window_stride): """See `sliding_window_batch` for details.""" self._input_dataset = input_dataset self._window_size = ops.convert_to_tensor( window_size, dtype=dtypes.int64, name="window_stride") self._window_stride = ops.convert_to_tensor( window_stride, dtype=dtypes.int64, name="window_stride") self._window_shift = ops.convert_to_tensor( window_shift, dtype=dtypes.int64, name="window_shift") input_structure = dataset_ops.get_structure(input_dataset) self._structure = input_structure._batch(None) # pylint: disable=protected-access variant_tensor = ged_ops.experimental_sliding_window_dataset( self._input_dataset._variant_tensor, # pylint: disable=protected-access window_size=self._window_size, window_shift=self._window_shift, window_stride=self._window_stride, **dataset_ops.flat_structure(self)) super(_SlideDataset, self).__init__(input_dataset, variant_tensor)