Contents

module Torch
  module NN
    module Utils
      def _single(value)
        _ntuple(1, value)
      end

      def _pair(value)
        _ntuple(2, value)
      end

      def _triple(value)
        _ntuple(3, value)
      end

      def _quadrupal(value)
        _ntuple(4, value)
      end

      def _ntuple(n, value)
        value.is_a?(Array) ? value : [value] * n
      end
    end
  end
end

Version data entries

30 entries across 30 versions & 1 rubygems

Version	Path
torch-rb-0.2.6	lib/torch/nn/utils.rb
torch-rb-0.2.5	lib/torch/nn/utils.rb
torch-rb-0.2.4	lib/torch/nn/utils.rb
torch-rb-0.2.3	lib/torch/nn/utils.rb
torch-rb-0.2.2	lib/torch/nn/utils.rb
torch-rb-0.2.1	lib/torch/nn/utils.rb
torch-rb-0.2.0	lib/torch/nn/utils.rb
torch-rb-0.1.8	lib/torch/nn/utils.rb
torch-rb-0.1.7	lib/torch/nn/utils.rb
torch-rb-0.1.6	lib/torch/nn/utils.rb