使用 Pytorch 张量创建简单的邻接矩阵答案

【问题标题】：Simple Adjacency matric creation with Pytorch Tensors使用 Pytorch 张量创建简单的邻接矩阵
【发布时间】：2021-07-28 14:41:14
【问题描述】：

我试图编写一个简单的函数来通过以下方式创建随机邻接矩阵：

def create_adj(a):
    a[a>0.5] = 1
    a[a<=0.5] = 0
    return a

假设a 被假定为torch.Tensor() 作为输入，但我收到以下错误：

TypeError: 'int' object does not support item assignment

如果我分开做事（即不在函数内部），我会做：

>> a = torch.rand(3,3)
>> a[a>0.5] = 1
>> a[a<=0.5] = 0

>> a
 tensor([[1., 1., 1.],
         [0., 0., 0.],
         [1., 0., 0.]])

但我不明白我在函数中做错了什么。

【问题讨论】：

【解决方案1】：

我假设您没有传递正确的变量 create_adj 函数。只要a 是torch.tensor，那么它应该可以工作。

或者，您可以直接使用掩码作为结果：

def create_adj(x): 
  return (a > .5).float()

【讨论】：