Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit 98f3dfa

Browse files
author
Seth Weidman
committed
Merge branch 'remove_tutorial_numbers' of github.com:pytorch/tutorials into remove_tutorial_numbers
2 parents 42e617d + 50f9908 commit 98f3dfa

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

beginner_source/chatbot_tutorial.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -537,7 +537,7 @@ def outputVar(l, voc):
537537
max_target_len = max([len(indexes) for indexes in indexes_batch])
538538
padList = zeroPadding(indexes_batch)
539539
mask = binaryMatrix(padList)
540-
mask = torch.ByteTensor(mask)
540+
mask = torch.BoolTensor(mask)
541541
padVar = torch.LongTensor(padList)
542542
return padVar, mask, max_target_len
543543

0 commit comments

Comments
 (0)