def __init__(self, hidden_dim):
    super(SketchPooler, self).__init__()
    self.fc = nn.Linear(hidden_dim, hidden_dim)
    self.activation = nn.Tanh()
def __init__(self, hidden_dim):
    super(SketchPooler, self).__init__()
    self.fc = nn.Linear(hidden_dim, hidden_dim)
    self.activation = nn.Tanh()