How to vectorize trace on an array of 2d matrices?...