pytorch get non diag element

Clever Cardinal answered on January 18, 2021 Popularity 1/10 Helpfulness 3/10

answer pytorch get non diag element

Comment

# 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)

Popularity 1/10 Helpfulness 3/10 Language python

Source: Grepper

Tags: element get python pytorch

Link to this answer
Share Copy Link

Contributed on Jan 18 2021

Clever Cardinal

0 Answers Avg Quality 2/10

Continue with Google

By continuing, I agree that I have read and agree to Greppers's Terms of Service and Privacy Policy.

Oops, You will need to install Grepper and log-in to perform this action.