pytorch中的跨层
Cross layer in pytorch
如何在pytorch中添加交叉层(https://arxiv.org/abs/1708.05123)。我想在我的神经网络中添加这样一个交叉层，它位于深层之前，以便更好地结合来自各种来源的我的特征。
谢谢！
您在 DeepCTR-Torch library. You can see the implementation here: https://github.com/shenweichen/DeepCTR-Torch/blob/master/deepctr_torch/models/dcn.py
中链接的论文中有 DCN 架构的 PyTorch 实现
如果你 pip install deepctr-torch 那么你可以只使用 CrossNet 图层作为 torch.Module:
from deepctr_torch.layers.interaction import CrossNet
crossnet = CrossNet(in_features=512)
如何在pytorch中添加交叉层(https://arxiv.org/abs/1708.05123)。我想在我的神经网络中添加这样一个交叉层，它位于深层之前，以便更好地结合来自各种来源的我的特征。
谢谢！
您在 DeepCTR-Torch library. You can see the implementation here: https://github.com/shenweichen/DeepCTR-Torch/blob/master/deepctr_torch/models/dcn.py
中链接的论文中有 DCN 架构的 PyTorch 实现如果你 pip install deepctr-torch 那么你可以只使用 CrossNet 图层作为 torch.Module:
from deepctr_torch.layers.interaction import CrossNet
crossnet = CrossNet(in_features=512)