specifyshape? should that be extract_diagonal?