pytorch Obtenha elemento não diagnóstico
# get diagonal elements in a matrix
torch.diag(a)
# get non diagonal elements in a matrix
a.flatten()[1:].view(n-1, n+1)[:,:-1].reshape(n, n-1)
Clever Cardinal