|
|
|
@@ -1,130 +0,0 @@
|
|
|
|
|
/**
|
|
|
|
|
* @file methods/ann/layer/layer_traits.hpp
|
|
|
|
|
* @author Marcus Edel
|
|
|
|
|
*
|
|
|
|
|
* This provides the LayerTraits class, a template class to get information
|
|
|
|
|
* about various layers.
|
|
|
|
|
*
|
|
|
|
|
* mlpack is free software; you may redistribute it and/or modify it under the
|
|
|
|
|
* terms of the 3-clause BSD license. You should have received a copy of the
|
|
|
|
|
* 3-clause BSD license along with mlpack. If not, see
|
|
|
|
|
* http://www.opensource.org/licenses/BSD-3-Clause for more information.
|
|
|
|
|
*/
|
|
|
|
|
#ifndef MLPACK_METHODS_ANN_LAYER_LAYER_TRAITS_HPP
|
|
|
|
|
#define MLPACK_METHODS_ANN_LAYER_LAYER_TRAITS_HPP
|
|
|
|
|
|
|
|
|
|
#include <mlpack/core/util/sfinae_utility.hpp>
|
|
|
|
|
|
|
|
|
|
namespace mlpack {
|
|
|
|
|
namespace ann {
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* This is a template class that can provide information about various layers.
|
|
|
|
|
* By default, this class will provide the weakest possible assumptions on
|
|
|
|
|
* layer, and each layer should override values as necessary. If a layer
|
|
|
|
|
* doesn't need to override a value, then there's no need to write a LayerTraits
|
|
|
|
|
* specialization for that class.
|
|
|
|
|
*/
|
|
|
|
|
template<typename LayerType>
|
|
|
|
|
class LayerTraits
|
|
|
|
|
{
|
|
|
|
|
public:
|
|
|
|
|
/**
|
|
|
|
|
* This is true if the layer is a binary layer.
|
|
|
|
|
*/
|
|
|
|
|
static const bool IsBinary = false;
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* This is true if the layer is an output layer.
|
|
|
|
|
*/
|
|
|
|
|
static const bool IsOutputLayer = false;
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* This is true if the layer is a bias layer.
|
|
|
|
|
*/
|
|
|
|
|
static const bool IsBiasLayer = false;
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
|
* This is true if the layer is a LSTM layer.
|
|
|
|
|
**/
|
|
|
|
|
static const bool IsLSTMLayer = false;
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
|
* This is true if the layer is a connection layer.
|
|
|
|
|
**/
|
|
|
|
|
static const bool IsConnection = false;
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
// This gives us a HasGradientCheck<T, U> type (where U is a function pointer)
|
|
|
|
|
// we can use with SFINAE to catch when a type has a Gradient(...) function.
|
|
|
|
|
HAS_MEM_FUNC(Gradient, HasGradientCheck);
|
|
|
|
|
|
|
|
|
|
// This gives us a HasDeterministicCheck<T, U> type (where U is a function
|
|
|
|
|
// pointer) we can use with SFINAE to catch when a type has a Deterministic()
|
|
|
|
|
// function.
|
|
|
|
|
HAS_MEM_FUNC(Deterministic, HasDeterministicCheck);
|
|
|
|
|
|
|
|
|
|
// This gives us a HasParametersCheck<T, U> type (where U is a function pointer)
|
|
|
|
|
// we can use with SFINAE to catch when a type has a Parameters() function.
|
|
|
|
|
HAS_MEM_FUNC(Parameters, HasParametersCheck);
|
|
|
|
|
|
|
|
|
|
// This gives us a HasAddCheck<T, U> type (where U is a function pointer) we
|
|
|
|
|
// can use with SFINAE to catch when a type has a Add() function.
|
|
|
|
|
HAS_MEM_FUNC(Add, HasAddCheck);
|
|
|
|
|
|
|
|
|
|
// This gives us a HasModelCheck<T> type we can use with SFINAE to catch when
|
|
|
|
|
// a type has a function named Model.
|
|
|
|
|
HAS_ANY_METHOD_FORM(Model, HasModelCheck);
|
|
|
|
|
|
|
|
|
|
// This gives us a HasLocationCheck<T, U> type (where U is a function pointer)
|
|
|
|
|
// we can use with SFINAE to catch when a type has a Location() function.
|
|
|
|
|
HAS_MEM_FUNC(Location, HasLocationCheck);
|
|
|
|
|
|
|
|
|
|
// This gives us a HasResetCheck<T, U> type (where U is a function pointer)
|
|
|
|
|
// we can use with SFINAE to catch when a type has a Reset() function.
|
|
|
|
|
HAS_MEM_FUNC(Reset, HasResetCheck);
|
|
|
|
|
|
|
|
|
|
// This gives us a HasResetCheck<T, U> type (where U is a function pointer)
|
|
|
|
|
// we can use with SFINAE to catch when a type has a ResetCell() function.
|
|
|
|
|
HAS_MEM_FUNC(ResetCell, HasResetCellCheck);
|
|
|
|
|
|
|
|
|
|
// This gives us a HasRewardCheck<T, U> type (where U is a function pointer) we
|
|
|
|
|
// can use with SFINAE to catch when a type has a Reward() function.
|
|
|
|
|
HAS_MEM_FUNC(Reward, HasRewardCheck);
|
|
|
|
|
|
|
|
|
|
// This gives us a HasInputWidth<T, U> type (where U is a function pointer) we
|
|
|
|
|
// can use with SFINAE to catch when a type has a InputWidth() function.
|
|
|
|
|
HAS_MEM_FUNC(InputWidth, HasInputWidth);
|
|
|
|
|
|
|
|
|
|
// This gives us a HasInputHeight<T, U> type (where U is a function pointer) we
|
|
|
|
|
// can use with SFINAE to catch when a type has a InputHeight() function.
|
|
|
|
|
HAS_MEM_FUNC(InputHeight, HasInputHeight);
|
|
|
|
|
|
|
|
|
|
// This gives us a HasRho<T, U> type (where U is a function pointer) we
|
|
|
|
|
// can use with SFINAE to catch when a type has a Rho() function.
|
|
|
|
|
HAS_MEM_FUNC(Rho, HasRho);
|
|
|
|
|
|
|
|
|
|
// This gives us a HasLoss<T, U> type (where U is a function pointer) we
|
|
|
|
|
// can use with SFINAE to catch when a type has a Loss() function.
|
|
|
|
|
HAS_MEM_FUNC(Loss, HasLoss);
|
|
|
|
|
|
|
|
|
|
// This gives us a HasRunCheck<T, U> type (where U is a function pointer) we
|
|
|
|
|
// can use with SFINAE to catch when a type has a Run() function.
|
|
|
|
|
HAS_MEM_FUNC(Run, HasRunCheck);
|
|
|
|
|
|
|
|
|
|
// This gives us a HasBiasCheck<T, U> type (where U is a function pointer) we
|
|
|
|
|
// can use with SFINAE to catch when a type has a Bias() function.
|
|
|
|
|
HAS_MEM_FUNC(Bias, HasBiasCheck);
|
|
|
|
|
|
|
|
|
|
// This gives us a HasMaxIterationsC<T, U> type (where U is a function pointer)
|
|
|
|
|
// we can use with SFINAE to catch when a type has a MaxIterations() function.
|
|
|
|
|
HAS_MEM_FUNC(MaxIterations, HasMaxIterations);
|
|
|
|
|
|
|
|
|
|
// This gives us a HasInShapeCheck<T> type we can use with SFINAE to catch when
|
|
|
|
|
// a type has a function named InputShape.
|
|
|
|
|
HAS_ANY_METHOD_FORM(InputShape, HasInputShapeCheck);
|
|
|
|
|
|
|
|
|
|
} // namespace ann
|
|
|
|
|
} // namespace mlpack
|
|
|
|
|
|
|
|
|
|
#endif
|
Ryan Curtin