Skip to content

Adding Filler to aidge_core

Cyril Moineau requested to merge Filler into dev

Context

This merge request aims at adding Filler/Initializer objects (name tdb).

The following filler will be implemented:

  • Constant
  • He
  • Normal
  • Uniform
  • Xavier
  • Add random number generator in Aidge

API example


import aidge_core
import numpy as np

t = aidge_core.Tensor(np.ones([1, 5, 1, 1]))
aidge_core.random.Generator.set_seed(1234) # set random seed
aidge_core.constant_filler(t, 5.0)
print("Constant")
print(t)
aidge_core.normal_filler(t)
print("Normal")
print(t)
aidge_core.uniform_filler(t, -1, 0)
print("Uniform")
print(t)
aidge_core.xavier_uniform_filler(t)
print("xavier_uniform")
print(t)
aidge_core.xavier_normal_filler(t)
print("xavier_normal")
print(t)
aidge_core.he_filler(t)
print("he")
print(t)
Edited by Cyril Moineau

Merge request reports

Loading