class TensorFlow::Data::Dataset

Attributes

output_shapes[R]

TODO remove

output_types[R]

TODO remove

Public Class Methods

from_tensor_slices(tensors) click to toggle source
# File lib/tensorflow/data/dataset.rb, line 21
def self.from_tensor_slices(tensors)
  TensorSliceDataset.new(tensors)
end
new(variant_tensor) click to toggle source
# File lib/tensorflow/data/dataset.rb, line 9
def initialize(variant_tensor)
  @variant_tensor = variant_tensor
end

Public Instance Methods

batch(batch_size, drop_remainder: false) click to toggle source
# File lib/tensorflow/data/dataset.rb, line 13
def batch(batch_size, drop_remainder: false)
  BatchDataset.new(self, batch_size, drop_remainder)
end
each() { |values| ... } click to toggle source
# File lib/tensorflow/data/dataset.rb, line 29
def each
  iterator, deleter = RawOps.anonymous_iterator_v2(output_types: @output_types, output_shapes: @output_shapes)
  RawOps.make_iterator(dataset: @variant_tensor, iterator: iterator)
  begin
    loop do
      values = RawOps.iterator_get_next_sync(iterator: iterator, output_types: @output_types, output_shapes: @output_shapes)
      yield values
    end
  rescue Error => e
    # iterate until end of sequence error
    raise e unless e.message == "End of sequence"
  end
ensure
  RawOps.delete_iterator(handle: iterator, deleter: deleter) if iterator
end
shuffle(buffer_size) click to toggle source
# File lib/tensorflow/data/dataset.rb, line 17
def shuffle(buffer_size)
  ShuffleDataset.new(self, buffer_size)
end
to_ptr() click to toggle source
# File lib/tensorflow/data/dataset.rb, line 25
def to_ptr
  @variant_tensor.to_ptr
end