pytorch

cd68083c - [functorch] Fixes _new_zeros_with_same_feature_meta batch rule (pytorch/functorch#603)

[functorch] Fixes _new_zeros_with_same_feature_meta batch rule (pytorch/functorch#603) The problem manifests itself when doing vmap over forward-mode AD in certain cases. Test Plan: - added a functorch-only OpInfo for this operator. - e2e tests