Sha256: 515510e1fc91935325f60d358f3792f9d12964e3b1e720bc44b46677180d2279
Contents?: true
Size: 471 Bytes
Versions: 54
Compression:
Stored size: 471 Bytes
Contents
module Torch
module NN
class LocalResponseNorm < Module
def initialize(size, alpha: 1e-4, beta: 0.75, k: 1.0)
super()
@size = size
@alpha = alpha
@beta = beta
@k = k
end
def forward(input)
F.local_response_norm(input, @size, alpha: @alpha, beta: @beta, k: @k)
end
def extra_inspect
format("%{size}, alpha: %{alpha}, beta: %{beta}, k: %{k}", **dict)
end
end
end
end
Version data entries
54 entries across 54 versions & 1 rubygems