def __init__(self, input_dim, hidden_dim):
    super(SketchEmbedding, self).__init__()
    self.embedding = nn.Linear(input_dim, hidden_dim)
def __init__(self, input_dim, hidden_dim):
    super(SketchEmbedding, self).__init__()
    self.embedding = nn.Linear(input_dim, hidden_dim)