Sha256: 9ff2fedc79b1c376343956b70137ee4d5891560dccfb13bdd2f0a107729171be
Contents?: true
Size: 554 Bytes
Versions: 55
Compression:
Stored size: 554 Bytes
Contents
module Torch
module NN
class RNN < RNNBase
def initialize(*args, **options)
if options.key?(:nonlinearity)
if options[:nonlinearity] == "tanh"
mode = "RNN_TANH"
elsif options[:nonlinearity] == "relu"
mode = "RNN_RELU"
else
raise ArgumentError, "Unknown nonlinearity: #{options[:nonlinearity]}"
end
options.delete(:nonlinearity)
else
mode = "RNN_TANH"
end
super(mode, *args, **options)
end
end
end
end
Version data entries
55 entries across 55 versions & 1 rubygems