Contents

module Torch
  module Utils
    module Data
      class TensorDataset
        def initialize(*tensors)
          @tensors = tensors
        end

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

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

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
torch-rb-0.1.2	lib/torch/utils/data/tensor_dataset.rb