Method: TensorFlow::Data::ShuffleDataset#initialize

Defined in:
lib/tensorflow/data/shuffle_dataset.rb

#initialize(input_dataset, buffer_size) ⇒ ShuffleDataset

Returns a new instance of ShuffleDataset.



4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
 
# File 'lib/tensorflow/data/shuffle_dataset.rb', line 4

def initialize(input_dataset, buffer_size)
  @input_dataset = input_dataset # keep reference for memory
  @output_types = input_dataset.output_types
  @output_shapes = input_dataset.output_shapes

  variant_tensor = RawOps.shuffle_dataset(
    input_dataset: input_dataset,
    buffer_size: TensorFlow.convert_to_tensor(buffer_size, dtype: :int64),
    seed: TensorFlow.convert_to_tensor(0, dtype: :int64),
    seed2: TensorFlow.convert_to_tensor(0, dtype: :int64),
    output_types: @output_types,
    output_shapes: @output_shapes
  )
  super(variant_tensor)
end