Deprecate apply_permutation

yzhangcs · May 11, 2023 · 42688d0 · 42688d0
1 parent bbe23fe
commit 42688d0
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/supar/modules/lstm.py b/supar/modules/lstm.py
@@ -7,7 +7,6 @@
 import torch
 import torch.nn as nn
 from supar.modules.dropout import SharedDropout
-from torch.nn.modules.rnn import apply_permutation
 from torch.nn.utils.rnn import PackedSequence, pack_padded_sequence
 
 
@@ -170,10 +169,7 @@ def permute_hidden(
     ) -> Tuple[torch.Tensor, torch.Tensor]:
         if permutation is None:
             return hx
-        h = apply_permutation(hx[0], permutation)
-        c = apply_permutation(hx[1], permutation)
-
-        return h, c
+        return hx[0].index_select(1, permutation), hx[1].index_select(1, permutation)
 
     def layer_forward(
         self,