1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
----------------------------------------------------------------------
--
-- Copyright (c) 2011 Clement Farabet, Koray Kavukcuoglu
--
-- Permission is hereby granted, free of charge, to any person obtaining
-- a copy of this software and associated documentation files (the
-- "Software"), to deal in the Software without restriction, including
-- without limitation the rights to use, copy, modify, merge, publish,
-- distribute, sublicense, and/or sell copies of the Software, and to
-- permit persons to whom the Software is furnished to do so, subject to
-- the following conditions:
--
-- The above copyright notice and this permission notice shall be
-- included in all copies or substantial portions of the Software.
--
-- THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
-- EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
-- MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
-- NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
-- LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
-- OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
-- WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
--
----------------------------------------------------------------------
-- description:
-- xlua - lots of new trainable modules that extend the nn
-- package.
--
-- history:
-- July 5, 2011, 8:51PM - import from Torch5 - Clement Farabet
----------------------------------------------------------------------
require 'torch'
require 'xlua'
require 'nn'
-- create global nnx table:
nnx = {}
-- c lib:
require 'libnnx'
-- for testing:
torch.include('nnx', 'jacobian.lua')
torch.include('nnx', 'test-all.lua')
torch.include('nnx', 'test-omp.lua')
-- tools:
torch.include('nnx', 'ConfusionMatrix.lua')
-- OpenMP module:
torch.include('nnx', 'OmpModule.lua')
-- pointwise modules:
torch.include('nnx', 'Abs.lua')
torch.include('nnx', 'Power.lua')
torch.include('nnx', 'Square.lua')
torch.include('nnx', 'Sqrt.lua')
torch.include('nnx', 'HardShrink.lua')
torch.include('nnx', 'Threshold.lua')
-- table-based modules:
torch.include('nnx', 'CMulTable.lua')
torch.include('nnx', 'CAddTable.lua')
torch.include('nnx', 'CDivTable.lua')
torch.include('nnx', 'CSubTable.lua')
-- reshapers:
torch.include('nnx', 'Narrow.lua')
torch.include('nnx', 'Replicate.lua')
-- spatial (images) operators:
torch.include('nnx', 'SpatialLinear.lua')
torch.include('nnx', 'SpatialLogSoftMax.lua')
torch.include('nnx', 'SpatialConvolutionTable.lua')
torch.include('nnx', 'SpatialMaxPooling.lua')
-- criterions:
torch.include('nnx', 'SuperCriterion.lua')
-- trainers:
torch.include('nnx', 'Trainer.lua')
torch.include('nnx', 'StochasticTrainer.lua')
-- datasets:
torch.include('nnx', 'DataSet.lua')
torch.include('nnx', 'DataList.lua')
|
