Python PyTorch OverArch用法及代碼示例

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

用法:

class torchrec.models.dlrm.OverArch(in_features: int, layer_sizes: List[int], device: Optional[torch.device] = None)

參數：

• in_features(int) -輸入的大小。

• layer_sizes(List[int]) -OverArch 的層大小。

• device(可選的[torch.device]) -默認計算設備。

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

Final Arch of DLRM - OverArch 上的簡單 MLP。

例子：

B = 20
D = 3
over_arch = OverArch(10, [5, 1])
logits = over_arch(torch.rand((B, 10)))