Class: Tensorflow::Data::OneShotIterator

Inherits:

Iterator

• Object
• Iterator
• Tensorflow::Data::OneShotIterator

Defined in:

lib/tensorflow/data/iterator.rb

Instance Attribute Summary

Attributes inherited from Iterator

#output_shapes, #output_types

Instance Method Summary

• #initialize(dataset) ⇒ OneShotIterator constructor

  A new instance of OneShotIterator.

Methods inherited from Iterator

from_structure, #get_next

Constructor Details

#initialize(dataset) ⇒ OneShotIterator

Returns a new instance of OneShotIterator.

# File 'lib/tensorflow/data/iterator.rb', line 21

def initialize(dataset)
  super(dataset.output_types, dataset.output_shapes)
  create_one_shot_iterator(dataset)
end