lib/torch.rb in torch-rb-0.15.0 vs lib/torch.rb in torch-rb-0.16.0
- old
+ new
@@ -121,9 +121,11 @@
require_relative "torch/nn/dropout2d"
require_relative "torch/nn/dropout3d"
require_relative "torch/nn/feature_alpha_dropout"
# nn activations
+require_relative "torch/nn/elu"
+require_relative "torch/nn/gelu"
require_relative "torch/nn/hardshrink"
require_relative "torch/nn/leaky_relu"
require_relative "torch/nn/log_sigmoid"
require_relative "torch/nn/prelu"
require_relative "torch/nn/relu"