def forward(self, hidden_states):
    pooled = self.fc1(hidden_states)
    pooled = self.fc2(pooled.view(pooled.size(0), (- 1)))
    return pooled
def forward(self, hidden_states):
    pooled = self.fc1(hidden_states)
    pooled = self.fc2(pooled.view(pooled.size(0), (- 1)))
    return pooled