def __init__(self, hidden_dim, M, cls_dim, max_length=250):
    super(SketchPooling, self).__init__()
    self.fc1 = nn.Linear(hidden_dim, 4)
    self.fc2 = nn.Linear((max_length * 4), cls_dim)
    self.re_fc = nn.Linear(hidden_dim, ((6 * M) + 3))
def __init__(self, hidden_dim, M, cls_dim, max_length=250):
    super(SketchPooling, self).__init__()
    self.fc1 = nn.Linear(hidden_dim, 4)
    self.fc2 = nn.Linear((max_length * 4), cls_dim)
    self.re_fc = nn.Linear(hidden_dim, ((6 * M) + 3))