fix test sentence synthesis

2026-09-09 11:16:00 +08:00 · 2021-12-09 13:35:43 +00:00
parent 20ac31dc71
commit 88d6399e12
1 changed files with 1 additions and 0 deletions
@@ -175,6 +175,7 @@ def embedding_to_torch(d_vector, cuda=False):
    if d_vector is not None:
        d_vector = np.asarray(d_vector)
        d_vector = torch.from_numpy(d_vector).type(torch.FloatTensor)
+        d_vector = d_vector.squeeze().unsqueeze(0)
    if cuda:
        return d_vector.cuda()
    return d_vector