问题描述:
在实现模型的过程中,移植的源码需要用到Pytorch提供的两个API的功能,即torch.nn.utils.rnn.pack_padded_sequence以及torch.nn.utils.rnn.pad_packed_sequenc。
不过好像没有在文档中找到相关API。。。
希望能得到帮助
【操作步骤&问题现象】
API 欠缺
【截图信息】
解答:
MindSpore的RNN系列API设计是不做pack和pad的,而是使用seq_length来传入padding前的batch中每条数据的实际长度,来保证padding的值不参与运算。