Answers for "pytorch get non diag element"

0

pytorch get non diag element

# 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)
Posted by: Guest on January-18-2021

Python Answers by Framework

Browse Popular Code Answers by Language