facebookresearch/fairseq
1
1
Fork 0
mirror of https://github.com/facebookresearch/fairseq.git synced 2024-11-11 03:31:32 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix #4708 (#4805)

Browse Source 
Co-authored-by: moslehpour <moslehpour@meta.com>
This commit is contained in:
Mohsen 2022-10-17 13:04:51 -07:00 committed by GitHub
parent 66d713b4d0
commit 05625e3e6e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
fairseq/modules/sinusoidal_positional_embedding.py
View File

@ -97,7 +97,7 @@ class SinusoidalPositionalEmbedding(nn.Module):
if self.onnx_trace:
flat_embeddings = self.weights.detach().index_select(0, positions.view(-1))
embedding_shape = torch.cat(
(bsz.view(1), seq_len.view(1), torch.tensor([-1], dtype=torch.long))
(bsz, seq_len, torch.tensor([-1], dtype=torch.long))
)
embeddings = torch.onnx.operators.reshape_from_tensor_shape(
flat_embeddings, embedding_shape