RockyML  0.0.1
A High-Performance Scientific Computing Framework
rocky::etna::mlp< T_e, T_layers_num, T_in_num, T_in_dim, T_out_dim, T_hidden_dim, T_opt_bias > Class Template Reference

Public Member Functions

void feed (T_e *layer_mem_ptr, T_e *in_mem_ptr, T_e *out_mem_ptr)
 apply the multi-layer perceptron on data in in_mem_ptr More...
 

Static Public Member Functions

static constexpr int deduce_num_params_in ()
 
static constexpr int deduce_num_params_hidden ()
 
static constexpr int deduce_num_params_out ()
 
static constexpr int deduce_num_params ()
 

Member Function Documentation

◆ feed()

template<typename T_e , int T_layers_num, int T_in_num, int T_in_dim, int T_out_dim, int T_hidden_dim, opt T_opt_bias = opt::bias>
void rocky::etna::mlp< T_e, T_layers_num, T_in_num, T_in_dim, T_out_dim, T_hidden_dim, T_opt_bias >::feed ( T_e *  layer_mem_ptr,
T_e *  in_mem_ptr,
T_e *  out_mem_ptr 
)
inline

apply the multi-layer perceptron on data in in_mem_ptr

Parameters
layer_mem_ptrmemory block containing layer parameters
in_mem_ptrmemory block containing input data
out_mem_ptrmemory block for storing the result
Returns
** void

The documentation for this class was generated from the following file: