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