torchfm.layer.
AttentionalFactorizationMachine
forward
x – Float tensor of size (batch_size, num_fields, embed_dim)
(batch_size, num_fields, embed_dim)
CompressedInteractionNetwork
CrossNetwork
FactorizationMachine
FeaturesEmbedding
x – Long tensor of size (batch_size, num_fields)
(batch_size, num_fields)
FeaturesLinear
FieldAwareFactorizationMachine
InnerProductNetwork
MultiLayerPerceptron
OuterProductNetwork