rixwew / pytorch-fm

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

One mistake in class MultiLayerPerceptron(torch.nn.Module) in layers.py #24

Closed BrandonCXY closed 4 years ago

BrandonCXY commented 4 years ago

你好,你对于Class MultiLayerPerceptron里的forward function中的x的注释错误 原注释::param x: Float tensor of size (batch_size, num_fields, embed_dim) 应改为: :param x: Float tensor of size (batch_size, embed_dim)

rixwew commented 4 years ago

Thank you! I fixed it.