def __init__(self, ip_size, op_size):
    super(gcn_layer, self).__init__()
    self.ip_size = ip_size
    self.op_size = op_size
    self.weights = Parameter(torch.rand(self.ip_size, self.op_size, dtype=torch.float32, requires_grad=True))
def __init__(self, ip_size, op_size):
    super(gcn_layer, self).__init__()
    self.ip_size = ip_size
    self.op_size = op_size
    self.weights = Parameter(torch.rand(self.ip_size, self.op_size, dtype=torch.float32, requires_grad=True))