pytorch
f1d7f001 - Batching rules for: torch.movedim, torch.narrow, Tensor.unfold (#42474)

Commit
4 years ago
Batching rules for: torch.movedim, torch.narrow, Tensor.unfold (#42474) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/42474 Test Plan: - `pytest test/test_vmap.py -v -k "Operators"` Reviewed By: ezyang Differential Revision: D22903513 Pulled By: zou3519 fbshipit-source-id: 06b3fb0c7d12b9a045c73a5c5a4f4e3207e07b02
Author
Parents
Loading