mask utils update for 0.4 cuda

2018-05-04 10:14:30 -07:00 · 2018-05-04 10:14:30 -07:00 · 6fbba8ef0f
parent c141726a96
commit 6fbba8ef0f
1 changed files with 1 additions and 1 deletions
--- a/utils.py
+++ b/utils.py
@ -5,7 +5,7 @@ import torch

 def get_mask_from_lengths(lengths):
    max_len = torch.max(lengths)
-    ids = torch.arange(0, max_len, out=torch.LongTensor(max_len)).cuda()
+    ids = torch.arange(0, max_len).long().cuda()
    mask = (ids < lengths.unsqueeze(1)).byte()
    return mask