Open
Description
Hi,
In layer.py, class FeaturesEmbedding
, one finds the line
self.offsets = np.array(0, *np.cumsum(field_dims)[:-1]), dtype=np.long)
My code generates an error. I replaced np.array
by torch.tensor
and np.cumsum
by torch.cumsum
, and the code works.
I believe the code as it is has a bug, which was fixed with my change. Could you please confirm this? I am not sure how the code can work as is. thanks.
Metadata
Metadata
Assignees
Labels
No labels