If you are given a collection of $n \\times n$ matrices say m of them, is there predefined functions in pytorch that performs a diagonal embedding on all of these into a lar
