[Torch5-devel] Concat in nn package
Status: Pre-Alpha
Brought to you by:
andresy
From: Ronan C. <ro...@co...> - 2007-11-28 20:44:10
|
Hey Jason pointed me out that Concat in nn package was not working. In fact, it was really buggy -- the implementation was not even finished. I corrected it, and improved it. l = nn.Concat([dim]) takes now an optional dim argument: the dimension along which you want to concatenate (default 1) [that is new, before it was 1 with no option]. each modules added in Concat (with l:add(...)) must take the same number of inputs. their outputs are concatenated (along dim given in the constructor) and returned by the Concat layer. you can feed tensors of any dimension to Concat (provided that the modules inside understand them). the outputed dimensions of each module might be different for the dimension "dim" but must (of course!) be the same in other dimensions. ronan. |