Class: Tensorflow::Data::ZipDataset

Inherits:
Dataset
  • Object
show all
Defined in:
lib/tensorflow/data/zip_dataset.rb

Constant Summary

Constants inherited from Dataset

Dataset::DEFAULT_READER_BUFFER_SIZE_BYTES

Instance Attribute Summary

Attributes inherited from Dataset

#output_shapes, #output_types, #variant_tensor

Instance Method Summary collapse

Methods inherited from Dataset

#batch, #data, #each, from_tensor_slices, from_tensors, #make_initializable_iterator, #make_one_shot_iterator, #map_func, #repeat, #shuffle, #to_ptr, to_tensor_array, #with_options

Constructor Details

#initialize(*datasets) ⇒ ZipDataset

Returns a new instance of ZipDataset.



4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
 
# 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