import data.examples.vector
open nat vector
set_option pp.implicit true
set_option pp.notation false
definition diag {A : Type} : Π {n}, vector (vector A n) n → vector A n,
diag nil := nil,
diag ((a :: va) :: vs) := a :: diag (map tail vs)