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)