def __init__(self, max_length, pos_hidden_dim):
    super(SketchLearnPositionEmbedding, self).__init__()
    print(max_length, pos_hidden_dim)
    self.pos_embedding = nn.Embedding(max_length, pos_hidden_dim)
def __init__(self, max_length, pos_hidden_dim):
    super(SketchLearnPositionEmbedding, self).__init__()
    print(max_length, pos_hidden_dim)
    self.pos_embedding = nn.Embedding(max_length, pos_hidden_dim)