分类：不定长序列补齐

2021
07-22

对pytorch中不定长序列补齐的操作

第二种方法通常是在load一个batch数据时,在collate_fn中进行补齐的.以下给出两种思路：第一种思路是比较容易想到的,就是对一个batch的样本进行遍历,然后使用np.pad对每一个样本进行补齐.forunitindata:mask=np.zeros(max_length)s_len=len(unit[0])#calculatethelengthofsequenceineachunitmask[:s_len]=1unit[0]=np.pad(unit[0],(0,max_length-s_len),'constant... 继续阅读 >