def __init__(self, hidden_dim, max_size, type_size, cls_in_input, rel_in_input):
    super(SketchDiscretePoolingModel, self).__init__()
    self.cls_in_input = cls_in_input
    self.rel_in_input = rel_in_input
    self.x_pooling = nn.Linear(hidden_dim, ((2 * max_size[0]) + 1))
    self.y_pooling = nn.Linear(hidden_dim, ((2 * max_size[1]) + 1))
    self.type_pooling = nn.Linear(hidden_dim, type_size)
def __init__(self, hidden_dim, max_size, type_size, cls_in_input, rel_in_input):
    super(SketchDiscretePoolingModel, self).__init__()
    self.cls_in_input = cls_in_input
    self.rel_in_input = rel_in_input
    self.x_pooling = nn.Linear(hidden_dim, ((2 * max_size[0]) + 1))
    self.y_pooling = nn.Linear(hidden_dim, ((2 * max_size[1]) + 1))
    self.type_pooling = nn.Linear(hidden_dim, type_size)