Ttclass/diag.m

Mimics the diag behavior for tensor train matrix. Syntax:

                           tt=diag(tt)

If tt is a square matrix, return a vector by computing diag of every core. If tt is a vector (one mode size is ones), return a diagonal matrix.