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