def __init__(self, shape, seed=0, seed2=0):
super(UniformIntNet, self).__init__()
self.shape = shape
self.seed = seed
self.seed2 = seed2
self.uniform_int = P.UniformInt(seed, seed2)
def __init__(self, shape, seed=0):
super(Net, self).__init__()
self.uniformint = P.UniformInt(seed=seed)
self.shape = shape