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