📌  相关文章
📜  pytorch 获取非诊断元素 - Python 代码示例

📅  最后修改于: 2022-03-11 14:45:30.901000             🧑  作者: Mango

代码示例1
# 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)