Skip to content

progress.md

Latest commit

 

History

History
196 lines (176 loc) · 4.16 KB

progress.md

File metadata and controls

196 lines (176 loc) · 4.16 KB
Raw

This list shows current supported modules and test status.

Our purpose is to facilitate the testing procedure. So backward functions won't be supported in a foresable future.

nn package

We follow the structure of nn package document. Blank cell indicates not implemented.

Containers

module code CPU test GPU test
Sequential √
Parallel
Concat √
DepthConcat √
Bottle

Transfer functions

module code CPU test GPU test
HardTanh
HardShrink
SoftShrink
SoftMax √
SoftMin
SoftPlus
SoftSign
LogSigmoid
LogSoftMax √
Sigmoid
Tanh √
ReLU √
ReLU6
PReLU
RReLU
ELU
LeakyReLU
SpatialSoftMax
AddConstant
MulConstant √

Simple layers

Parameterized Modules

module code CPU test GPU test
Linear √
SparseLinear
BiLinear
PartialLinear
Add √
Mul
CMul
Euclidean
WeightedEuclidean
Cosine

Modules that adapt basic Tensor methods

module code CPU test GPU test
Copy
Narrow
Replicate
Reshape √
View √
Contiguous
Select
MaskedSelect
Index
Squeeze
Unsqueeze
Transpose

Modules that adapt mathematical Tensor methods

module code CPU test GPU test
AddConstant
MulConstant √
Max
Min
Mean
Sum
Exp
Log
Abs
Power
Square
Sqrt √
Clamp
Normalize √
MM

Miscellaneous Modules

module code CPU test GPU test
BatchNormalization √
Identity
Dropout
SpatialDropout
VolumetricDropout
Padding
L1Penalty
GradientReversal
GPU
TemporalDynamicKMaxPooling

Table layers

table Container Modules encapsulate sub-Modules

module code CPU test GPU test
ConcatTable
ParallelTable
MapTable

Table Conversion Modules convert between tables and Tensors or tables

module code CPU test GPU test
SplitTable
JoinTable
MixtureTable
SelectTable
NarrowTable
FlattenTable

Pair Modules compute a measure like distance or similarity from a pair (table) of input Tensors

module code CPU test GPU test
PairwiseDistance
DotProduct
CosineDistance

CMath Modules perform element-wise operations on a table of Tensors

module code CPU test GPU test
CAddTable
CSubTable
CMulTable
CDivTable
CMaxTable
CMinTable

Table of Criteria

module code CPU test GPU test
CriterionTable

Convolution layers

Temporal Modules

module code CPU test GPU test
TemporalConvolution
TemporalSubSampling
TemporalMaxPooling
LookupTable

Spatial Modules

module code CPU test GPU test
SpatialConvolution √
SpatialConvolutionMM √
SpatialFullConvolution
SpatialDilatedConvolution
SpatialConvolutionLocal
SpatialSubSampling
SpatialMaxPooling √
SpatialDilatedMaxPooling
SpatialFractionalMaxPooling
SpatialAveragePooling √
SpatialAdaptvieMaxPooling
SpatialMaxUnpooling
SpatialLPPooling √
SpatialConvolutionMap
SpatialZeroPadding
SpatialReflectionPadding √
SpatialReplicationPadding
SpatialSubtractiveNormalization
SpatialCrossMapLRN √
SpatialBatchNormalization √
SpatialUpsamplingNearest
SpatialUpsamplingBilinear

Volumetric Modules

module code CPU test GPU test
VolumetricConvolution
VolumetricFullConvolution
VolumetricDilatedConvolution
VolumetricMaxPooling
VolumetricDilatedMaxPooling
VolumetricAveragePooling
VolumetricMaxUnpooling
VolumetricReplicationPadding

dpnn package

Currently only Inception and Decorator module are supported.