torch-rb 0.1.2 → 0.1.3

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

@@ -3,6 +3,9 @@ module Torch
     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
 

data/lib/torch/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: torch-rb
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - Andrew Kane
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-11-27 00:00:00.000000000 Z
+date: 2019-11-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rice
@@ -110,7 +110,15 @@ files:
 - 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
@@ -119,7 +127,17 @@ files:
 - 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
@@ -127,7 +145,7 @@ files:
 - lib/torch/version.rb
 homepage: https://github.com/ankane/torch-rb
 licenses:
-- MIT
+- BSD-3-Clause
 metadata: {}
 post_install_message: 
 rdoc_options: []