class Tensorflow::Data::ReinitializableIterator
Public Class Methods
new(output_types, output_shapes, shared_name: '')
click to toggle source
Calls superclass method
Tensorflow::Data::Iterator::new
# File lib/tensorflow/data/iterator.rb, line 60 def initialize(output_types, output_shapes, shared_name: '') super(output_types, output_shapes) create_iterator_from_structure(shared_name) end
Public Instance Methods
make_initializer(dataset)
click to toggle source
# File lib/tensorflow/data/iterator.rb, line 65 def make_initializer(dataset) RawOps.make_iterator(dataset.variant_tensor, @iterator) end
Private Instance Methods
create_iterator_from_structure(shared_name)
click to toggle source
# File lib/tensorflow/data/iterator.rb, line 71 def create_iterator_from_structure(shared_name) @iterator = RawOps.iterator_v2(shared_name: shared_name, output_types: self.output_types, output_shapes: self.output_shapes) end