Files
Flux/include/random/detail/binomial_serial.h
T
Bausager eb0a49591e Dropout Layer
Implemented rng::uniform and rng::binomial for single values, vectors and matrices. implemeted dropout layers and tested it. Also fixed the validation code. Before it used y one place, now it uses y_test as it should.
2026-05-21 15:38:49 +02:00

46 lines
1.0 KiB
C++

#pragma once
#include "random/engine.h"
#include <random>
//#include <type_traits>
#include "utils/matrix.h"
#include "utils/vector.h"
namespace rng::detail {
template <typename T>
inline T binomial_serial(const uint64_t trials, const double probability) {
std::binomial_distribution<int> dist(trials, probability);
return static_cast<T>(dist(rng::engine()));
}
template <typename T>
utils::Vector<T> binomial_serial(const uint64_t size, const uint64_t trials, const double probability){
utils::Vector<T> v(size);
for (uint64_t i = 0; i < size; ++i){
v[i] = binomial_serial<T>(trials, probability);
}
return v;
}
template <typename T>
utils::Matrix<T> binomial_serial(const uint64_t rows, const uint64_t cols, const uint64_t trials, const double probability){
utils::Matrix<T> A(rows, cols);
for (uint64_t i = 0; i < rows; ++i){
for (uint64_t j = 0; j < cols; ++j){
A(i,j) = binomial_serial<T>(trials, probability);
}
}
return A;
}
}