Sha256: aecd149c8956cdfdd66512b40970c411830c7a592731cf06e87f4beb644511da
Contents?: true
Size: 309 Bytes
Versions: 56
Compression:
Stored size: 309 Bytes
Contents
module Torch
module NN
class ReLU < Module
def initialize(inplace: false)
super()
@inplace = inplace
end
def forward(input)
F.relu(input, inplace: @inplace)
end
def extra_inspect
@inplace ? "inplace: true" : ""
end
end
end
end
Version data entries
56 entries across 56 versions & 1 rubygems