Python PyTorch OverArch用法及代碼示例

本文簡要介紹python語言中 torchrec.models.deepfm.OverArch 的用法。

用法:

class torchrec.models.deepfm.OverArch(in_features: int)

參數：

in_features(int) -交互拱的輸出維度。

基礎：torch.nn.modules.module.Module

Final Arch - 簡單的 MLP。輸出隻是一個目標。

例子：

B = 20
over_arch = OverArch()
logits = over_arch(torch.rand((B, 10)))