Python PyTorch InteractionArch用法及代码示例

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

用法:

class torchrec.models.dlrm.InteractionArch(sparse_feature_names: List[str])

参数：

sparse_feature_names(List[str]) -尺寸 F

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

处理SparseArch (sparse_features) 和DenseArch (dense_features) 的输出。返回每个稀疏特征对的成对点积，每个稀疏特征与密集层输出的点积，以及密集层本身(全部连接)。

注意

dense_features (D) 的维度预计与 sparse_features 的维度匹配，以便可以计算它们之间的点积。

例子：

D = 3
B = 10
keys = ["f1", "f2"]
F = len(keys)
inter_arch = InteractionArch(sparse_feature_names=keys)

dense_features = torch.rand((B, D))

sparse_features = KeyedTensor(
   keys=keys,
   length_per_key=[D, D],
   values=torch.rand((B, D * F)),
)

#  B X (D + F + F choose 2)
concat_dense = inter_arch(dense_features, sparse_features)