如何解决关于如何有效创建此矩阵/蒙版的任何想法?
[![我想有效地制作一个移动窗口为1s的矩阵的火炬张量或numpy。
因此,例如下面的矩阵将是window = 3。 对角线元素的右边有3 1s,左边有3 1s,但是它不会像循环矩阵那样环绕,因此第1行只有4 1s。
任何人都有任何想法,可以用作遮罩。
解决方法
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
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。