关于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]

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