pytorch

c71dfa6e - [functorch] add diag_embed batch rule