EmbeddingBag w/ differentiable per_sample_weights (#18957)
authorRichard Zou <zou3519@gmail.com>
Wed, 10 Apr 2019 01:09:01 +0000 (18:09 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 10 Apr 2019 01:13:06 +0000 (18:13 -0700)
commit447d74a0747be7686dc44bb9db57ca45954fc865
treed8e245f6ad8d6e346dc0913a1929413e889b7b87
parentc889ff6cf8c18017490c7ed0e99a524fc5ff7422
EmbeddingBag w/ differentiable per_sample_weights (#18957)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18957
ghimport-source-id: 7396ca08b137ea40f04285764a9d9a6d4f19227e

Reviewed By: cpuhrsch

Differential Revision: D14856526

Pulled By: zou3519

fbshipit-source-id: 949faea219c7c02ad981b1db610a477194d3f5c9
aten/src/ATen/native/EmbeddingBag.cpp
aten/src/ATen/native/cuda/EmbeddingBag.cu
aten/src/ATen/native/native_functions.yaml
aten/src/TH/THBlasUtils.h
test/test_nn.py
tools/autograd/derivatives.yaml