Sha256: 2a4042b5424ebde9d74d4ac3196d5111c27e5349c08e91dd1ed7da09ef6e9fbf
Contents?: true
Size: 388 Bytes
Versions: 56
Compression:
Stored size: 388 Bytes
Contents
module Torch
module NN
class Softplus < Module
def initialize(beta: 1, threshold: 20)
super()
@beta = beta
@threshold = threshold
end
def forward(input)
F.softplus(input, beta: @beta, threshold: @threshold)
end
def extra_inspect
format("beta: %s, threshold: %s", @beta, @threshold)
end
end
end
end
Version data entries
56 entries across 56 versions & 1 rubygems