rixwew / pytorch-fm

Factorization Machine models in PyTorch
MIT License
1.04k stars 225 forks source link

dimension issue #2

Closed AmazingDD closed 5 years ago

AmazingDD commented 5 years ago

In the forward process, the dimension of input X in all models are (batch_size, field_size), here can I treat all these field size as feature size?