Class: Torch::Utils::Data::TensorDataset

Inherits:

Object

• Object
• Torch::Utils::Data::TensorDataset

Defined in:

lib/torch/utils/data/tensor_dataset.rb

Instance Method Summary

• #[](index) ⇒ Object
• #initialize(*tensors) ⇒ TensorDataset constructor

  A new instance of TensorDataset.

• #size ⇒ Object

Constructor Details

#initialize(*tensors) ⇒ TensorDataset

Returns a new instance of TensorDataset.

# File 'lib/torch/utils/data/tensor_dataset.rb', line 5

def initialize(*tensors)
  @tensors = tensors
end

Instance Method Details

#[](index) ⇒ Object

# File 'lib/torch/utils/data/tensor_dataset.rb', line 9

def [](index)
  @tensors.map { |t| t[index] }
end

#size ⇒ Object

# File 'lib/torch/utils/data/tensor_dataset.rb', line 13

def size
  @tensors[0].size(0)
end