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