frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-25 08:11:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update sparse.py (#2336)

Browse Source
This commit is contained in:
adampolyak
2017-08-08 14:16:52 +03:00
committed by Soumith Chintala
parent 5caa42b538
commit e51fec3be0
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/nn/_functions/thnn/sparse.py
View File

@ -45,7 +45,7 @@ class Embedding(Function):
output = weight.new()
if max_norm is not None:
cls._renorm(indices, weight, max_norm, norm_type)
cls._renorm(ctx, indices, weight, max_norm, norm_type)
if indices.dim() == 1:
output = torch.index_select(weight, 0, indices)