class Tensorflow::Data::ZipDataset

Public Class Methods

new(*datasets) click to toggle source
Calls superclass method Tensorflow::Data::Dataset::new
# File lib/tensorflow/data/zip_dataset.rb, line 4
def initialize(*datasets)
  datasets = datasets.flatten(1)
  tensors = Array.new(datasets.size)
  output_types = Array.new(datasets.size)
  output_shapes = Array.new(datasets.size)

  datasets.each_with_index do |dataset, i|
    tensors[i] = dataset.variant_tensor
    output_types[i] = dataset.output_types
    output_shapes[i] = dataset.output_shapes
  end

  @output_types = output_types.flatten
  @output_shapes = output_shapes.flatten(1)
  variant_tensor = RawOps.zip_dataset(tensors, n: tensors.count, output_types: @output_types, output_shapes: @output_shapes)

  super(variant_tensor)
end