class Tensorflow::Data::RepeatDataset

Public Class Methods

new(dataset, count) click to toggle source
Calls superclass method Tensorflow::Data::Dataset::new
# File lib/tensorflow/data/repeat_dataset.rb, line 4
def initialize(dataset, count)
  @output_types = dataset.output_types
  @output_shapes = dataset.output_shapes

  variant_tensor = RawOps.repeat_dataset(dataset.variant_tensor, Tensor.new(count, dtype: :int64),
                                         output_types: @output_types,
                                         output_shapes: @output_shapes)

  super(variant_tensor)
end