fix `Tensors must have same number of dimensions: got 1 and 2` in Cro…

shenweichen / DeepCTR-Torch

【PyTorch】Easy-to-use,Modular and Extendible package of deep-learning based CTR models.

https://deepctr-torch.readthedocs.io/en/latest/index.html

Apache License 2.0

2.95k stars 696 forks source link

fix `Tensors must have same number of dimensions: got 1 and 2` in Cro… #291

Open T-baby opened 6 months ago

T-baby commented 6 months ago

Fix the error "Tensors must have the same number of dimensions: got 1 and 2" in CrossNetMix. The function torch.squeeze(x_l) generates a one-dimensional vector.