class Chainer::Datasets::TupleDataset
Public Class Methods
new(*datasets)
click to toggle source
# File lib/chainer/datasets/tuple_dataset.rb, line 4 def initialize(*datasets) if datasets.empty? raise "no datasets are given" end length = datasets[0].shape[0] datasets.each_with_index do |dataset, idx| raise "dataset of the index #{idx} has a wrong length" unless dataset.shape[0] == length end @datasets = datasets @length = length end
Public Instance Methods
[](index)
click to toggle source
# File lib/chainer/datasets/tuple_dataset.rb, line 18 def [](index) batches = @datasets.map do |dataset| dataset.ndim > 1 ? dataset[index, false] : dataset[index] end if index.kind_of?(Enumerable) length = batches[0].shape[0] length.times.map {|i| batches.map { |m| m.ndim > 1 ? m[i, false] : m[i] } } else batches end end
size()
click to toggle source
# File lib/chainer/datasets/tuple_dataset.rb, line 30 def size @length end