docs/initialization_8hpp_source.html

 #pragma once
 #include <dlprim/context.hpp>
 #include <dlprim/tensor.hpp>
 namespace dlprim {
     class RandomState;

     void set_to_zero(Tensor &t,ExecutionContext const &e);
     void set_to_constant(Tensor &t,double value,ExecutionContext const &e);

     void set_to_urandom(Tensor &t,RandomState &state,float minv,float maxv,ExecutionContext const &e);

     void set_to_normal(Tensor &t,RandomState &state,float mean,float sigma,ExecutionContext const &e);
     void set_to_bernoulli(Tensor &t,RandomState &state,float p,ExecutionContext const &e);

 }
dlprim::set_to_normal
void set_to_normal(Tensor &t, RandomState &state, float mean, float sigma, ExecutionContext const &e)
set t values to normal distribution with mean and sigma), seed is updated

dlprim::set_to_zero
void set_to_zero(Tensor &t, ExecutionContext const &e)
Set value of t to zero.

dlprim::set_to_urandom
void set_to_urandom(Tensor &t, RandomState &state, float minv, float maxv, ExecutionContext const &e)
set t values to uniform random values in range [minv,maxv), seed is updated

dlprim::set_to_bernoulli
void set_to_bernoulli(Tensor &t, RandomState &state, float p, ExecutionContext const &e)
set t values to bernully distribution with mean and sigma), seed is updated

dlprim
Mane namespace.
Definition: context.hpp:9

dlprim::set_to_constant
void set_to_constant(Tensor &t, double value, ExecutionContext const &e)
Set to constant, value is casted to t.dtype()