/
vuron
/
adept
Обзор
Документация
Войти
/
vuron
/
adept
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
src/nn/init.cpp
22 строки
483 B
kolkir
BatchNorm2d module interface and tensor scalar filling
16 фев 2025, 22:27
16 фев 2025, 22:27
47a1bbe
Код
Авторство
О чём код?
#include <adept/nn/init.hpp> #include "../dispatch/tensor_init.hpp" namespace adept { void fill_zero(Tensor& tensor) { fill_zero_tensor(tensor); } void fill(Tensor& tensor, const std::any& scalar) { fill_tensor(tensor, scalar); } void fill_uniform(float32_t start, float32_t end, Tensor& tensor) { fill_uniform_tensor(start, end, tensor); } void fill_normal(float32_t mean, float32_t std, Tensor& tensor) { fill_normal_tensor(mean, std, tensor); } } // namespace adept