vuron/adept: Платформа для обучения и вывода нейронных сетей, с API приближенным к PyTorch.

src/nn/init.cpp

22 строки483 B

kolkir

BatchNorm2d module interface and tensor scalar filling

16 фев 2025, 22:27

16 фев 2025, 22:2747a1bbe

#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