[Fix] transpose use numpy

10 months ago · 39ec5dc5e7
parent 67ae7c8dd2
commit 39ec5dc5e7
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)