Doc(Language.ANY, DocScope.ALL){
            """
                Extract the diagonal part from the input array.
                If input is
                [ 1, 0, 0]