class Tensorflow::Data::InitializableIterator
Attributes
initializer[R]
Public Class Methods
new(dataset, shared_name: '')
click to toggle source
Calls superclass method
Tensorflow::Data::Iterator::new
# File lib/tensorflow/data/iterator.rb, line 46 def initialize(dataset, shared_name: '') super(dataset.output_types, dataset.output_shapes) create_initializable_iterator(dataset, shared_name) end
Private Instance Methods
create_initializable_iterator(dataset, shared_name)
click to toggle source
# File lib/tensorflow/data/iterator.rb, line 53 def create_initializable_iterator(dataset, shared_name) @iterator = RawOps.iterator_v2(shared_name: shared_name, output_types: self.output_types, output_shapes: self.output_shapes) @initializer = RawOps.make_iterator(dataset.variant_tensor, @iterator) end