本文介绍了PyTorch 张量的零对角线?的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
限时送ChatGPT账号..是否有一种简单方法可以将 PyTorch 张量的对角线归零?
Is there a simple way to zero the diagonal of a PyTorch tensor?
例如我有:
tensor([[2.7183, 0.4005, 2.7183, 0.5236],
[0.4005, 2.7183, 0.4004, 1.3469],
[2.7183, 0.4004, 2.7183, 0.5239],
[0.5236, 1.3469, 0.5239, 2.7183]])
我想得到:
tensor([[0.0000, 0.4005, 2.7183, 0.5236],
[0.4005, 0.0000, 0.4004, 1.3469],
[2.7183, 0.4004, 0.0000, 0.5239],
[0.5236, 1.3469, 0.5239, 0.0000]])
推荐答案
我相信最简单的方法是使用 torch.diagonal
:
I believe the simplest would be to use torch.diagonal
:
z = torch.randn(4,4)
torch.diagonal(z, 0).zero_()
print(z)
>>> tensor([[ 0.0000, -0.6211, 0.1120, 0.8362],
[-0.1043, 0.0000, 0.1770, 0.4197],
[ 0.7211, 0.1138, 0.0000, -0.7486],
[-0.5434, -0.8265, -0.2436, 0.0000]])
通过这种方式,代码非常明确,您可以将性能委托给 pytorch 的内置函数.
This way, the code is perfectly explicit, and you delegate the performance to pytorch's built in functions.
这篇关于PyTorch 张量的零对角线?的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!
更多推荐
[db:关键词]
发布评论