关于DIEN的PyTorch实现

[Q-Qing]

你好，我在阅读DIEN的PyTorch代码时发现一个地方可能存在错误
在Interest Extractor Layer中，GRU的隐藏层应该保留前T-1个用于计算辅助loss
但是DIEN代码的152行是
gru_embed=pad_interests[:,1:]
这样是不是取了后T-1个hidden state?
我理解的应该是
gru_embed=pad_interests[:,:-1]

[Prayforhanluo]

我印象中好像是最后存的是最晚的时刻，最新的时刻存在了首位。有时间看看。