torch-rb 0.1.2 → 0.1.3
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/CHANGELOG.md +10 -0
- data/LICENSE.txt +46 -22
- data/README.md +14 -5
- data/ext/torch/ext.cpp +248 -31
- data/lib/torch.rb +80 -9
- data/lib/torch/ext.bundle +0 -0
- data/lib/torch/inspector.rb +4 -3
- data/lib/torch/nn/alpha_dropout.rb +9 -0
- data/lib/torch/nn/conv2d.rb +12 -24
- data/lib/torch/nn/convnd.rb +41 -0
- data/lib/torch/nn/dropout.rb +9 -0
- data/lib/torch/nn/dropout2d.rb +9 -0
- data/lib/torch/nn/dropout3d.rb +9 -0
- data/lib/torch/nn/dropoutnd.rb +15 -0
- data/lib/torch/nn/embedding.rb +52 -0
- data/lib/torch/nn/feature_alpha_dropout.rb +9 -0
- data/lib/torch/nn/functional.rb +54 -12
- data/lib/torch/nn/linear.rb +2 -2
- data/lib/torch/nn/module.rb +30 -0
- data/lib/torch/optim/adadelta.rb +57 -0
- data/lib/torch/optim/adagrad.rb +71 -0
- data/lib/torch/optim/adam.rb +81 -0
- data/lib/torch/optim/adamax.rb +68 -0
- data/lib/torch/optim/adamw.rb +82 -0
- data/lib/torch/optim/asgd.rb +65 -0
- data/lib/torch/optim/lr_scheduler/lr_scheduler.rb +33 -0
- data/lib/torch/optim/lr_scheduler/step_lr.rb +17 -0
- data/lib/torch/optim/optimizer.rb +56 -0
- data/lib/torch/optim/rmsprop.rb +76 -0
- data/lib/torch/optim/rprop.rb +68 -0
- data/lib/torch/optim/sgd.rb +48 -16
- data/lib/torch/tensor.rb +38 -4
- data/lib/torch/utils/data/data_loader.rb +10 -4
- data/lib/torch/utils/data/tensor_dataset.rb +3 -0
- data/lib/torch/version.rb +1 -1
- metadata +21 -3
data/lib/torch/version.rb
CHANGED
metadata
CHANGED
|
@@ -1,14 +1,14 @@
|
|
|
1
1
|
--- !ruby/object:Gem::Specification
|
|
2
2
|
name: torch-rb
|
|
3
3
|
version: !ruby/object:Gem::Version
|
|
4
|
-
version: 0.1.
|
|
4
|
+
version: 0.1.3
|
|
5
5
|
platform: ruby
|
|
6
6
|
authors:
|
|
7
7
|
- Andrew Kane
|
|
8
8
|
autorequire:
|
|
9
9
|
bindir: bin
|
|
10
10
|
cert_chain: []
|
|
11
|
-
date: 2019-11-
|
|
11
|
+
date: 2019-11-30 00:00:00.000000000 Z
|
|
12
12
|
dependencies:
|
|
13
13
|
- !ruby/object:Gem::Dependency
|
|
14
14
|
name: rice
|
|
@@ -110,7 +110,15 @@ files:
|
|
|
110
110
|
- lib/torch.rb
|
|
111
111
|
- lib/torch/ext.bundle
|
|
112
112
|
- lib/torch/inspector.rb
|
|
113
|
+
- lib/torch/nn/alpha_dropout.rb
|
|
113
114
|
- lib/torch/nn/conv2d.rb
|
|
115
|
+
- lib/torch/nn/convnd.rb
|
|
116
|
+
- lib/torch/nn/dropout.rb
|
|
117
|
+
- lib/torch/nn/dropout2d.rb
|
|
118
|
+
- lib/torch/nn/dropout3d.rb
|
|
119
|
+
- lib/torch/nn/dropoutnd.rb
|
|
120
|
+
- lib/torch/nn/embedding.rb
|
|
121
|
+
- lib/torch/nn/feature_alpha_dropout.rb
|
|
114
122
|
- lib/torch/nn/functional.rb
|
|
115
123
|
- lib/torch/nn/init.rb
|
|
116
124
|
- lib/torch/nn/linear.rb
|
|
@@ -119,7 +127,17 @@ files:
|
|
|
119
127
|
- lib/torch/nn/parameter.rb
|
|
120
128
|
- lib/torch/nn/relu.rb
|
|
121
129
|
- lib/torch/nn/sequential.rb
|
|
130
|
+
- lib/torch/optim/adadelta.rb
|
|
131
|
+
- lib/torch/optim/adagrad.rb
|
|
132
|
+
- lib/torch/optim/adam.rb
|
|
133
|
+
- lib/torch/optim/adamax.rb
|
|
134
|
+
- lib/torch/optim/adamw.rb
|
|
135
|
+
- lib/torch/optim/asgd.rb
|
|
136
|
+
- lib/torch/optim/lr_scheduler/lr_scheduler.rb
|
|
137
|
+
- lib/torch/optim/lr_scheduler/step_lr.rb
|
|
122
138
|
- lib/torch/optim/optimizer.rb
|
|
139
|
+
- lib/torch/optim/rmsprop.rb
|
|
140
|
+
- lib/torch/optim/rprop.rb
|
|
123
141
|
- lib/torch/optim/sgd.rb
|
|
124
142
|
- lib/torch/tensor.rb
|
|
125
143
|
- lib/torch/utils/data/data_loader.rb
|
|
@@ -127,7 +145,7 @@ files:
|
|
|
127
145
|
- lib/torch/version.rb
|
|
128
146
|
homepage: https://github.com/ankane/torch-rb
|
|
129
147
|
licenses:
|
|
130
|
-
-
|
|
148
|
+
- BSD-3-Clause
|
|
131
149
|
metadata: {}
|
|
132
150
|
post_install_message:
|
|
133
151
|
rdoc_options: []
|