class Tensorflow::Data::MapDataset

Public Class Methods

new(input_dataset, function, other_arguments: []) click to toggle source
Calls superclass method Tensorflow::Data::Dataset::new
# File lib/tensorflow/data/map_dataset.rb, line 4
def initialize(input_dataset, function, other_arguments: [])
  @output_types = function.output_types
  @output_shapes = function.output_shapes

  variant_tensor = RawOps.map_dataset(input_dataset.variant_tensor, other_arguments,
                                      f: function,
                                      output_types: @output_types,
                                      output_shapes: @output_shapes)

  super(variant_tensor)
end