torch-rb 0.1.3

data/lib/torch/utils/data/data_loader.rb

@@ -0,0 +1,27 @@
+module Torch
+  module Utils
+    module Data
+      class DataLoader
+        include Enumerable
+
+        attr_reader :dataset
+
+        def initialize(dataset, batch_size: 1)
+          @dataset = dataset
+          @batch_size = batch_size
+        end
+
+        def each
+          size.times do |i|
+            start_index = i * @batch_size
+            yield @dataset[start_index...(start_index + @batch_size)]
+          end
+        end
+
+        def size
+          (@dataset.size / @batch_size.to_f).ceil
+        end
+      end
+    end
+  end
+end

data/lib/torch/utils/data/tensor_dataset.rb

@@ -0,0 +1,22 @@
+module Torch
+  module Utils
+    module Data
+      class TensorDataset
+        def initialize(*tensors)
+          unless tensors.all? { |t| t.size(0) == tensors[0].size(0) }
+            raise Error, "Tensors must all have same dim 0 size"
+          end
+          @tensors = tensors
+        end
+
+        def [](index)
+          @tensors.map { |t| t[index] }
+        end
+
+        def size
+          @tensors[0].size(0)
+        end
+      end
+    end
+  end
+end

data/lib/torch/version.rb

@@ -0,0 +1,3 @@
+module Torch
+  VERSION = "0.1.3"
+end

metadata

@@ -0,0 +1,169 @@
+--- !ruby/object:Gem::Specification
+name: torch-rb
+version: !ruby/object:Gem::Version
+  version: 0.1.3
+platform: ruby
+authors:
+- Andrew Kane
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2019-11-30 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rice
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake-compiler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '5'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '5'
+- !ruby/object:Gem::Dependency
+  name: numo-narray
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: 
+email: andrew@chartkick.com
+executables: []
+extensions:
+- ext/torch/extconf.rb
+extra_rdoc_files: []
+files:
+- CHANGELOG.md
+- LICENSE.txt
+- README.md
+- ext/torch/ext.cpp
+- ext/torch/extconf.rb
+- lib/torch-rb.rb
+- lib/torch.rb
+- lib/torch/ext.bundle
+- lib/torch/inspector.rb
+- lib/torch/nn/alpha_dropout.rb
+- lib/torch/nn/conv2d.rb
+- lib/torch/nn/convnd.rb
+- lib/torch/nn/dropout.rb
+- lib/torch/nn/dropout2d.rb
+- lib/torch/nn/dropout3d.rb
+- lib/torch/nn/dropoutnd.rb
+- lib/torch/nn/embedding.rb
+- lib/torch/nn/feature_alpha_dropout.rb
+- lib/torch/nn/functional.rb
+- lib/torch/nn/init.rb
+- lib/torch/nn/linear.rb
+- lib/torch/nn/module.rb
+- lib/torch/nn/mse_loss.rb
+- lib/torch/nn/parameter.rb
+- lib/torch/nn/relu.rb
+- lib/torch/nn/sequential.rb
+- lib/torch/optim/adadelta.rb
+- lib/torch/optim/adagrad.rb
+- lib/torch/optim/adam.rb
+- lib/torch/optim/adamax.rb
+- lib/torch/optim/adamw.rb
+- lib/torch/optim/asgd.rb
+- lib/torch/optim/lr_scheduler/lr_scheduler.rb
+- lib/torch/optim/lr_scheduler/step_lr.rb
+- lib/torch/optim/optimizer.rb
+- lib/torch/optim/rmsprop.rb
+- lib/torch/optim/rprop.rb
+- lib/torch/optim/sgd.rb
+- lib/torch/tensor.rb
+- lib/torch/utils/data/data_loader.rb
+- lib/torch/utils/data/tensor_dataset.rb
+- lib/torch/version.rb
+homepage: https://github.com/ankane/torch-rb
+licenses:
+- BSD-3-Clause
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '2.4'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.0.3
+signing_key: 
+specification_version: 4
+summary: Deep learning for Ruby, powered by LibTorch
+test_files: []