2021
07-22
07-22
解决pytorch rnn 变长输入序列的问题
pytorch实现变长输入的rnn分类输入数据是长度不固定的序列数据,主要讲解两个部分1、Data.DataLoader的collate_fn用法,以及按batch进行padding数据2、pack_padded_sequence和pad_packed_sequence来处理变长序列collate_fnDataloader的collate_fn参数,定义数据处理和合并成batch的方式。由于pack_padded_sequence用到的tensor必须按照长度从大到小排过序的,所以在Collate_fn中,需要完成两件事,一是把当前batch的样本按照当前ba...
继续阅读 >