def __init__(self, max_size, type_size, hidden_dim, pool_type):
    super(SketchDiscreteEmbedding, self).__init__()
    self.x_embedding = nn.Embedding(((2 * max_size[0]) + 2), (hidden_dim // 2))
    self.y_embedding = nn.Embedding(((2 * max_size[1]) + 2), (hidden_dim // 2))
    self.type_embedding = nn.Embedding((type_size + 1), hidden_dim)
    assert (pool_type in ['sum', 'con'])
    self.pool_type = pool_type
def __init__(self, max_size, type_size, hidden_dim, pool_type):
    super(SketchDiscreteEmbedding, self).__init__()
    self.x_embedding = nn.Embedding(((2 * max_size[0]) + 2), (hidden_dim // 2))
    self.y_embedding = nn.Embedding(((2 * max_size[1]) + 2), (hidden_dim // 2))
    self.type_embedding = nn.Embedding((type_size + 1), hidden_dim)
    assert (pool_type in ['sum', 'con'])
    self.pool_type = pool_type