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