malarinv/tacotron2
1
0
Fork 0
mirror of https://github.com/malarinv/tacotron2 synced 2026-03-08 01:32:35 +00:00
Code Issues Releases Wiki Activity

Merge pull request #4 from NVIDIA/mask-utils-0.4

Browse Source 
mask utils update for 0.4 cuda
This commit is contained in:
Rafael Valle
2018-05-04 11:02:30 -07:00
committed by GitHub
parent c141726a96 6fbba8ef0f
commit 2c545ac800
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
utils.py
View File

@@ -5,7 +5,7 @@ import torch
def get_mask_from_lengths(lengths): def get_mask_from_lengths(lengths):
max_len = torch.max(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() mask = (ids < lengths.unsqueeze(1)).byte()
return mask return mask