[Fix] transpose use numpy (#3933)

PaddlePaddle · Dec 5, 2024 · ff539ef · ff539ef
1 parent c0fafd0
commit ff539ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddlespeech/s2t/models/wav2vec2/modules/wav2vec2_model.py b/paddlespeech/s2t/models/wav2vec2/modules/wav2vec2_model.py
@@ -1267,7 +1267,7 @@ def __init__(self, deconstruct_idx=None):
     def forward(self, x):
         if self.deconstruct_idx is not None:
             x = x[self.deconstruct_idx]
-        trans_dim = paddle.arange(x.dim())
+        trans_dim = np.arange(x.dim())
         trans_dim[-1], trans_dim[-2] = trans_dim[-2], trans_dim[-1]
         return x.transpose(trans_dim)