如何沿对角线替换 PyTorch 张量中的特定值？

问题描述

比如有一个PyTorch矩阵A：

A = tensor([[3,2,1],[1,2],[2,0]])

我需要将对角线上的 0 替换为 1，所以结果应该是：

tensor([[3,1,1]])

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）