Forward AD formulas batch 3 (#58094)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/58094
Test Plan: Imported from OSS
Reviewed By: zou3519
Differential Revision: D28387762
Pulled By: albanD
fbshipit-source-id: fc395c92af7ebb5ebae95c40f6c76273047f4097
