class Tensorflow::Data::ShuffleDataset
Public Class Methods
new(input_dataset, buffer_size)
click to toggle source
Calls superclass method
Tensorflow::Data::Dataset::new
# File lib/tensorflow/data/shuffle_dataset.rb, line 4 def initialize(input_dataset, buffer_size) @input_dataset = input_dataset @output_types = input_dataset.output_types @output_shapes = input_dataset.output_shapes buffer_size = Tensor.new(buffer_size, dtype: :int64) seed = Tensor.new(::Random.rand(100_000_000), dtype: :int64) seed2 = Tensor.new(::Random.rand(100_000_000), dtype: :int64) variant_tensor = RawOps.shuffle_dataset(input_dataset.variant_tensor, buffer_size, seed, seed2, output_types: @output_types, output_shapes: @output_shapes) super(variant_tensor) end