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