pytorch
1cd5ba49 - Add batching rule for "is_complex", "conj" (#44649)

Commit
4 years ago
Add batching rule for "is_complex", "conj" (#44649) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/44649 To unblock #43208, which adds "is_complex" checks to backward formulas that are being tested for batched gradient support with vmap. Test Plan: - `pytest test/test_vmap.py -v` Reviewed By: anjali411 Differential Revision: D23685356 Pulled By: zou3519 fbshipit-source-id: 29e41a9296336f6d1008e3040cade4c643bf5ebf
Author
Parents
Loading