def forward(self, hidden_states):
    hidden_states = hidden_states.to(next(self.fc.parameters()).device)
    inter_states = self.fc(hidden_states.contiguous())
    inter_states = self.activation(inter_states)
    return inter_states
def forward(self, hidden_states):
    hidden_states = hidden_states.to(next(self.fc.parameters()).device)
    inter_states = self.fc(hidden_states.contiguous())
    inter_states = self.activation(inter_states)
    return inter_states