Contents

module Torch
  module NN
    class MaxUnpool1d < MaxUnpoolNd
      def initialize(kernel_size, stride: nil, padding: 0)
        super()
        @kernel_size = _single(kernel_size)
        @stride = _single(stride || kernel_size)
        @padding = _single(padding)
      end

      def forward(input, indices, output_size: nil)
        F.max_unpool1d(input, indices, @kernel_size, stride: @stride, padding: @padding, output_size: output_size)
      end
    end
  end
end

Version data entries

54 entries across 54 versions & 1 rubygems

Version	Path
torch-rb-0.3.2	lib/torch/nn/max_unpool1d.rb
torch-rb-0.3.1	lib/torch/nn/max_unpool1d.rb
torch-rb-0.3.0	lib/torch/nn/max_unpool1d.rb
torch-rb-0.2.7	lib/torch/nn/max_unpool1d.rb
torch-rb-0.2.6	lib/torch/nn/max_unpool1d.rb
torch-rb-0.2.5	lib/torch/nn/max_unpool1d.rb
torch-rb-0.2.4	lib/torch/nn/max_unpool1d.rb
torch-rb-0.2.3	lib/torch/nn/max_unpool1d.rb
torch-rb-0.2.2	lib/torch/nn/max_unpool1d.rb
torch-rb-0.2.1	lib/torch/nn/max_unpool1d.rb
torch-rb-0.2.0	lib/torch/nn/max_unpool1d.rb
torch-rb-0.1.8	lib/torch/nn/max_unpool1d.rb
torch-rb-0.1.7	lib/torch/nn/max_unpool1d.rb
torch-rb-0.1.6	lib/torch/nn/max_unpool1d.rb