[Fix] transpose use numpy (#3933)

3 weeks ago · ff539ef007
parent c0fafd0647
commit ff539ef007
1 changed files with 1 additions and 1 deletions
--- a/paddlespeech/s2t/models/wav2vec2/modules/wav2vec2_model.py
+++ b/paddlespeech/s2t/models/wav2vec2/modules/wav2vec2_model.py
@ -1267,7 +1267,7 @@ class TransposeLast(nn.Layer):
    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)