|
| Conv2dLayer (op::Operation< T > *input, const std::vector< unsigned int > &filter_shape={3, 3}, int out_channels=1, const std::vector< unsigned int > &padding={0, 0}, const std::vector< unsigned int > &strides={1, 1}, const std::vector< unsigned int > &dilation_rates={1, 1}, bool use_cross_correlation=true, bool use_bias=false, tensor_filler_t< T > filter_initializer={GLOROT,{0.0, 0.2f}}, tensor_filler_t< T > bias_initializer={GLOROT,{0.0, 0.2f}}) |
|
virtual std::vector< op::Operation< T > * > | get_weights () |
|
op::Operation< T > * | get_filter () |
|
op::Operation< T > * | get_bias () |
|
virtual op::Operation< T > * | out () |
|
op::Operation< T > * | get_input () |
|
op::Operation< T > * | get_output () |
|
std::vector< unsigned int > | get_input_shape () const |
|
std::vector< unsigned int > | get_output_shape () const |
|
unsigned int | get_input_shape (unsigned int i) const |
|
unsigned int | get_output_shape (unsigned int i) const |
|
void | set_name (std::string name) |
|
std::string | get_name () const |
|
|
void | init (const std::vector< unsigned int > &filter_shape) |
|
| Layer (std::vector< unsigned int > input_shape, op::Operation< T > *input) |
|
|
Tensor< T > * | filter_tensor |
|
Tensor< T > * | bias_tensor |
|
op::Operation< T > * | filter |
|
op::Operation< T > * | bias |
|
int | in_channels |
|
int | out_channels |
|
bool | use_cross_correlation |
|
bool | use_bias |
|
int | pad_h |
|
int | pad_w |
|
int | stride_h |
|
int | stride_w |
|
int | dilation_h |
|
int | dilation_w |
|
tensor_filler_t< T > | filter_initializer |
|
tensor_filler_t< T > | bias_initializer |
|
std::vector< unsigned int > | input_shape |
|
std::vector< unsigned int > | output_shape |
|
op::Operation< T > * | input |
|
op::Operation< T > * | output |
|
std::string | name |
|
The documentation for this class was generated from the following files: