Sha256: 78c9ee26069c088e9041162cf81c9898b22bf9823d4ad182b23dcd137f5c222a
Contents?: true
Size: 347 Bytes
Versions: 56
Compression:
Stored size: 347 Bytes
Contents
module Torch
module NN
class ConstantPadNd < Module
def initialize(value)
super()
@value = value
end
def forward(input)
F.pad(input, @padding, mode: "constant", value: @value)
end
def extra_inspect
format("padding: %s, value: %s", @padding, @value)
end
end
end
end
Version data entries
56 entries across 56 versions & 1 rubygems