class Tensorflow::Data::TfRecordDataset
Constants
- DEFAULT_BUFFER_SIZE
Public Class Methods
new(filenames, compression_type='', buffer_size=DEFAULT_BUFFER_SIZE)
click to toggle source
Calls superclass method
Tensorflow::Data::Dataset::new
# File lib/tensorflow/data/tf_record_dataset.rb, line 6 def initialize(filenames, compression_type='', buffer_size=DEFAULT_BUFFER_SIZE) filenames = Array(filenames) @output_types = [:string] @output_shapes = [[]] buffer_size = Tensor.new(buffer_size, dtype: :int64) if buffer_size variant_tensor = RawOps.tf_record_dataset(filenames, compression_type, buffer_size) super(variant_tensor) end