PyTorch 张量的零对角线? [英] Zero diagonal of a PyTorch tensor?

问题描述

是否有一种简单方法可以将 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屋！