Sha256: 97ef3ca4b040bf5514120a36d656c574801d99fe785ff402b8a218d36c61b56e
Contents?: true
Size: 416 Bytes
Versions: 56
Compression:
Stored size: 416 Bytes
Contents
module Torch
module NN
class PReLU < Module
def initialize(num_parameters: 1, init: 0.25)
@num_parameters = num_parameters
super()
@weight = Parameter.new(Tensor.new(num_parameters).fill!(init))
end
def forward(input)
F.prelu(input, @weight)
end
def extra_inspect
format("num_parameters: %s", @num_parameters)
end
end
end
end
Version data entries
56 entries across 56 versions & 1 rubygems