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