projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: ad22804) | patch
[1.10] Remove torch.vmap (#65496)
authorRichard Zou <zou3519@users.noreply.github.com>
Fri, 24 Sep 2021 17:29:08 +0000 (13:29 -0400)
committerGitHub <noreply@github.com>
Fri, 24 Sep 2021 17:29:08 +0000 (10:29 -0700)
commit0e857bf1092e0245925f6de596f7cb38de88cc23
tree68b0ea601699cabe2734f2d7fdb2d3ba708a9e31tree | snapshot
parentad22804b952afde13e92fce321b310764b2147d1commit | diff
[1.10] Remove torch.vmap (#65496)

torch.vmap is a prototype feature and should not be in the stable
binary. This PR:
- Removes the torch.vmap API
- Removes the documentation entry for torch.vmap
- Changes the vmap tests to use an internal API instead of torch.vmap.

Test Plan:
- Tested locally (test_torch, test_autograd, test_type_hints, test_vmap),
but also wait for CI.
docs/source/torch.rst diff | blob | history
test/test_vmap.py diff | blob | history
torch/__init__.py diff | blob | history
Domain: Machine Learning / ML Framework;