mlpack
Static Public Member Functions | List of all members
FFTConvolution< BorderMode, padLastDim > Class Template Reference

Computes the two-dimensional convolution through fft. More...

Static Public Member Functions

template
<
typename
eT
,
typename
Border
=
BorderMode
>
static std::enable_if< std::is_same< Border, ValidConvolution >::value, void >::type Convolution (const arma::Mat< eT > &input, const arma::Mat< eT > &filter, arma::Mat< eT > &output)
 
template
<
typename
eT
,
typename
Border
=
BorderMode
>
static std::enable_if< std::is_same< Border, FullConvolution >::value, void >::type Convolution (const arma::Mat< eT > &input, const arma::Mat< eT > &filter, arma::Mat< eT > &output)
 
template
<
typename
eT
>
static void Convolution (const arma::Cube< eT > &input, const arma::Cube< eT > &filter, arma::Cube< eT > &output)
 
template
<
typename
eT
>
static void Convolution (const arma::Mat< eT > &input, const arma::Cube< eT > &filter, arma::Cube< eT > &output)
 
template
<
typename
eT
>
static void Convolution (const arma::Cube< eT > &input, const arma::Mat< eT > &filter, arma::Cube< eT > &output)
 

Detailed Description


template<typename BorderMode = FullConvolution, const bool padLastDim = false>
class mlpack::ann::FFTConvolution< BorderMode, padLastDim >

Computes the two-dimensional convolution through fft.

This class allows specification of the type of the border type. The convolution can be computed with the valid border type or the full border type (default).

FullConvolution: returns the full two-dimensional convolution. ValidConvolution: returns only those parts of the convolution that are computed without the zero-padded edges.

Template Parameters
BorderModeType of the border mode (FullConvolution or ValidConvolution).
padLastDimPad the last dimension of the input to to turn it from odd to even.

Definition at line 37 of file fft_convolution.hpp.

Member Function Documentation

◆ Convolution() [1/5]

static std::enable_if< std::is_same<Border, ValidConvolution>::value, void>::type Convolution ( const arma::Mat< eT > &  input,
const arma::Mat< eT > &  filter,
arma::Mat< eT > &  output 
)
inlinestatic

Definition at line 54 of file fft_convolution.hpp.

Referenced by SVDConvolution< BorderMode >::Convolution(), and FFTConvolution< BorderMode, padLastDim >::Convolution().

◆ Convolution() [2/5]

static std::enable_if< std::is_same<Border, FullConvolution>::value, void>::type Convolution ( const arma::Mat< eT > &  input,
const arma::Mat< eT > &  filter,
arma::Mat< eT > &  output 
)
inlinestatic

Definition at line 89 of file fft_convolution.hpp.

◆ Convolution() [3/5]

static void Convolution ( const arma::Cube< eT > &  input,
const arma::Cube< eT > &  filter,
arma::Cube< eT > &  output 
)
inlinestatic

Definition at line 135 of file fft_convolution.hpp.

References FFTConvolution< BorderMode, padLastDim >::Convolution().

◆ Convolution() [4/5]

static void Convolution ( const arma::Mat< eT > &  input,
const arma::Cube< eT > &  filter,
arma::Cube< eT > &  output 
)
inlinestatic

Definition at line 166 of file fft_convolution.hpp.

References FFTConvolution< BorderMode, padLastDim >::Convolution().

◆ Convolution() [5/5]

static void Convolution ( const arma::Cube< eT > &  input,
const arma::Mat< eT > &  filter,
arma::Cube< eT > &  output 
)
inlinestatic

Definition at line 194 of file fft_convolution.hpp.

References FFTConvolution< BorderMode, padLastDim >::Convolution().

The documentation for this class was generated from the following file:
  • /home/ryan/src/mlpack.org/_src/mlpack-git/src/mlpack/methods/ann/convolution_rules/fft_convolution.hpp