3 Commits

Author	SHA1	Message	Date
Anjali Chourdia	a37be201c1	Implement torch.nn.Embedding / EmbeddingBag in PyTorch C++ API (#26358) ... Summary: added more variables to EmbeddingOptions and updated EmbeddingImpl reset, forward functions. Also added EmbeddingBag. ----- This PR is BC-breaking in the following way: Previously, `EmbeddingOptions` supports `count` and `dimension` as options arguments. After this PR, they are renamed to `num_embeddings` and `embedding_dim` respectively. Pull Request resolved: https://github.com/pytorch/pytorch/pull/26358 Differential Revision: D17714337 Pulled By: yf225 fbshipit-source-id: f9f969c68e4bece106b92f8e2e02ac39c8455fb7	2019-10-08 22:13:39 -07:00
ShahriarSS	cad3abb036	Adding ModuleList to modules.h (#25346) ... Summary: Here is a PR adding ```ModuleList``` to ```modules.h``` so that it can be used by including ```torch/torch.h```. yf225 edit: Closes https://github.com/pytorch/pytorch/issues/25293. Pull Request resolved: https://github.com/pytorch/pytorch/pull/25346 Differential Revision: D17115013 Pulled By: yf225 fbshipit-source-id: 38a1848b9a8272fa411865dfc83b76d10c5789a0	2019-08-29 10:49:22 -07:00
ShahriarSS	cd622f7655	C++ ModuleList ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/24317 Differential Revision: D16893634 Pulled By: yf225 fbshipit-source-id: 9d810ad5a41bd46e5d0fa766e851178c60226866	2019-08-19 10:02:40 -07:00