28#include "tiny_dnn/util/util.h"
31namespace weight_init {
35 virtual void fill(vec_t *weight, serial_size_t
fan_in, serial_size_t
fan_out) = 0;
61 void fill(vec_t *weight, serial_size_t
fan_in, serial_size_t
fan_out)
override {
80 void fill(vec_t *weight, serial_size_t
fan_in, serial_size_t
fan_out)
override {
81 CNN_UNREFERENCED_PARAMETER(
fan_out);
94 void fill(vec_t *weight, serial_size_t
fan_in, serial_size_t
fan_out)
override {
95 CNN_UNREFERENCED_PARAMETER(
fan_in);
96 CNN_UNREFERENCED_PARAMETER(
fan_out);
98 gaussian_rand(weight->begin(), weight->end(),
float_t(0), scale_);
107 void fill(vec_t *weight, serial_size_t
fan_in, serial_size_t
fan_out)
override {
108 CNN_UNREFERENCED_PARAMETER(
fan_in);
109 CNN_UNREFERENCED_PARAMETER(
fan_out);
111 std::fill(weight->begin(), weight->end(), scale_);
120 void fill(vec_t *weight, serial_size_t
fan_in, serial_size_t
fan_out)
override {
121 CNN_UNREFERENCED_PARAMETER(
fan_out);
125 gaussian_rand(weight->begin(), weight->end(),
float_t(0),
sigma);
Simple image utility class.
Definition image.h:94
Definition weight_init.h:102
Definition weight_init.h:33
Definition weight_init.h:89
Definition weight_init.h:115
Use fan-in(number of input weight for each neuron) for scaling.
Definition weight_init.h:75
Definition weight_init.h:38
Use fan-in and fan-out for scaling.
Definition weight_init.h:56