CPU forward gives `IndexError` for diagonal matrices

[ekhahniii]

Applying Sym3Eig to diagonal matrices gives IndexError: select(): index 1 out of range for tensor of size [1, 3] at dimension 0.

Simple reproduction:

m = torch.diag(torch.rand(3)).unsqueeze(0)
e, v = Sym3Eig.apply(m)