def __init__(self):
    super().__init__()
    self.param = flow.nn.Parameter(flow.Tensor(init_value, device=flow.device(device)))
def __init__(self):
    super().__init__()
    self.param = flow.nn.Parameter(flow.Tensor(init_value, device=flow.device(device)))