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)))