Python PyTorch DenseArch用法及代码示例

本文简要介绍python语言中 torchrec.models.dlrm.DenseArch 的用法。

用法:

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

参数：

• in_features(int) -密集输入特征的维度。

• layer_sizes(List[int]) -层大小列表。

• device(可选的[torch.device]) -默认计算设备。

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

处理 DLRM 模型的密集特征。

例子：

B = 20
D = 3
dense_arch = DenseArch(10, layer_sizes=[15, D])
dense_embedded = dense_arch(torch.rand((B, 10)))