Class: Tensorflow::Data::ReinitializableIterator

Inherits:
Iterator
  • Object
show all
Defined in:
lib/tensorflow/data/iterator.rb

Instance Attribute Summary

Attributes inherited from Iterator

#output_shapes, #output_types

Instance Method Summary collapse

Methods inherited from Iterator

from_structure, #get_next

Constructor Details

#initialize(output_types, output_shapes, shared_name: '') ⇒ ReinitializableIterator

Returns a new instance of ReinitializableIterator.



60
61
62
63
 
# 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

Instance Method Details

#make_initializer(dataset) ⇒ Object 



65
66
67
 
# File 'lib/tensorflow/data/iterator.rb', line 65

def make_initializer(dataset)
  RawOps.make_iterator(dataset.variant_tensor, @iterator)
end