关于如何有效创建此矩阵/蒙版的任何想法？

问题描述

[！[我想有效地制作一个移动窗口为1s的矩阵的火炬张量或numpy。

因此，例如下面的矩阵将是window = 3。 对角线元素的右边有3 1s，左边有3 1s，但是它不会像循环矩阵那样环绕，因此第1行只有4 1s。

任何人都有任何想法，可以用作遮罩。

] 1] 1

解决方法

Pytorch提供了tensor.diagonal方法，该方法使您可以访问张量的任何对角线。要将值分配给张量的结果视图，可以使用tensor.copy_。那会给你类似的东西：

def circulant(n,window):
    circulant_t = torch.zeros(n,n)
    # [0,1,2,...,window,-1,-2,window]
    offsets = [0] + [i for i in range(window)] + [-i for i in range(window)]
    for offset in offsets:
        #size of the 1-tensor depends on the length of the diagonal
        circulant_t.diagonal(offset=offset).copy_(torch.ones(n-abs(offset)))
    return circulant_t