Exemple #1
0
 def __init__(self, shape, seed=0, seed2=0):
     super(PoissonNet, self).__init__()
     self.shape = shape
     self.seed = seed
     self.seed2 = seed2
     self.poisson = P.Poisson(seed, seed2)
Exemple #2
0
 def __init__(self, shape, seed=0, seed2=0):
     super(Net, self).__init__()
     self.poisson = P.Poisson()
     self.shape = shape