关于pytorch张量计算

Question

我想问一个关于火炬计算的问题，如果我只想减去矩阵对角线上的元素而不改变剩余位置的元素，有没有办法实现呢？

Answer 1

做到这一点的一种方法是获取对角线，对其元素进行所需的操作并替换原始元素。示例代码：

x = torch.rand(3, 3)
#get the original diagonal and for example substract 3
replaced_diag = x.diagonal() - 3
#replace the original diagonal
x.diagonal().copy_(replaced_diag)

参考看看这个：Replace diagonal elements with vector in PyTorch