Sha256: 11140861200b791ad5247e6952aebfac6bb51a3955038fd291fef0e84530762d
Contents?: true
Size: 400 Bytes
Versions: 56
Compression:
Stored size: 400 Bytes
Contents
module Torch
module NN
class Sequential < Module
def initialize(*args)
super()
# TODO support hash arg (named modules)
args.each_with_index do |mod, idx|
add_module(idx.to_s, mod)
end
end
def forward(input)
@modules.values.each do |mod|
input = mod.call(input)
end
input
end
end
end
end
Version data entries
56 entries across 56 versions & 1 rubygems