Possibility of adding DICE to replace prelu? Also, a small bug for the GPU implementation for newer PyTorch versions

[YingHH1]

Hi，

谢谢大佬的code。想问一下在DIN/DIEN里是否会在未来加入DICE激活函数来取代torch自身提供的prelu?

目前发现的小问题，在gpu运行下(torch 1.9.0)： 在DIEN.py里因为pytorch版本更新带来的一些new behaviour，需要加入 keys_length=keys_length.to(torch.device("cpu")) 作为torch里的改动要求（需要非cuda的cpu long type）。之后在interest.py里加入 keys_length=keys_length.to(torch.device("cuda" if torch.cuda.is_available() else "cpu")) 来改回为GPU。

非常感谢！