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)
def __init__(self, shape, seed=0, seed2=0):
super(Net, self).__init__()
self.poisson = P.Poisson()
self.shape = shape