rocky::etna::linear< T_e, T_in_num, T_in_dim, T_out_dim, T_opt_bias > Class Template Reference

base class for static layers More...

#include <linear.h>

Public Member Functions
void	feed (T_e *layer_mem_ptr, T_e *in_mem_ptr, T_e *out_mem_ptr)

Static Public Member Functions
static constexpr int	deduce_num_params_weights ()
static constexpr int	deduce_num_params_bias ()
static constexpr int	deduce_num_params ()

Detailed Description

template<typename T_e, int T_in_num, int T_in_dim, int T_out_dim, opt T_opt_bias = opt::bias>
class rocky::etna::linear< T_e, T_in_num, T_in_dim, T_out_dim, T_opt_bias >

base class for static layers

Member Function Documentation

feed()

template<typename T_e , int T_in_num, int T_in_dim, int T_out_dim, opt T_opt_bias = opt::bias>

void rocky::etna::linear< T_e, T_in_num, T_in_dim, T_out_dim, T_opt_bias >::feed ( T_e *  layer_mem_ptr, T_e *  in_mem_ptr, T_e *  out_mem_ptr  )

inline

(T_in_num * T_in_dim) @ (T_in_dim x T_out_dim) -> (T_in_num * T_out_dim)

---

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

• include/rocky/etna/linear.h