From 07b7beabad05bd300f964d53d9368debeefefc4f Mon Sep 17 00:00:00 2001 From: blessyyyu <954793264@qq.com> Date: Wed, 23 Mar 2022 18:32:01 +0800 Subject: [PATCH] remove files --- learnkws/learn_mask.py | 30 ------------------------------ learnkws/test_learn.py | 13 ------------- 2 files changed, 43 deletions(-) delete mode 100644 learnkws/learn_mask.py delete mode 100644 learnkws/test_learn.py diff --git a/learnkws/learn_mask.py b/learnkws/learn_mask.py deleted file mode 100644 index 4627ebd..0000000 --- a/learnkws/learn_mask.py +++ /dev/null @@ -1,30 +0,0 @@ -''' -Date: 2022-03-04 18:10:52 -LastEditors: Cyan -LastEditTime: 2022-03-07 10:21:34 -''' - -import torch - - -def padding_mask(lengths: torch.Tensor) -> torch.Tensor: - """ - Examples: - >>> lengths = torch.tensor([2, 2, 3], dtype=torch.int32) - >>> mask = padding_mask(lengths) - >>> print(mask) - tensor([[False, False, True], - [False, False, True], - [False, False, False]]) - """ - batch_size = lengths.size(0) - max_len = int(lengths.max().item()) - seq = torch.arange(max_len, dtype=torch.int64, device=lengths.device) - seq = seq.expand(batch_size, max_len) - return seq >= lengths.unsqueeze(1) - -if __name__ == '__main__': - lengths = torch.tensor([2, 2, 3], dtype=torch.int32) - print(lengths.numel()) - mask = padding_mask(lengths) - print(mask, mask.size()) diff --git a/learnkws/test_learn.py b/learnkws/test_learn.py deleted file mode 100644 index df1f7c7..0000000 --- a/learnkws/test_learn.py +++ /dev/null @@ -1,13 +0,0 @@ -''' -Date: 2022-03-04 18:10:52 -LastEditors: Cyan -LastEditTime: 2022-03-07 10:21:34 -''' - -if __name__ == '__main__': - a = [1,2,3,4,5,6,7] - for i in range(len(a)): - print('i = ', i) - if a[i] >= 3: - i += 2 - # print('a[i] = ' , a[i])