将字符串矩阵转换为 PyTorch 张量答案

【问题标题】：Converting matrix of strings to PyTorch tensor将字符串矩阵转换为 PyTorch 张量
【发布时间】：2023-01-23 19:38:18
【问题描述】：

我想将以下矩阵转换为 PyTorch 张量：

[['SELF', '', '', '', ''],
 ['nsubj', 'SELF', '', '', ''],
 ['', 'compound', 'SELF', '', ''],
 ['dobj', '', '', 'SELF', ''],
 ['pobj', '', '', '', 'SELF']]

我想要一个布尔矩阵，其中除 empty 之外的任何字符串位置都会有一个 1，否则为 0。这应该很容易，但我似乎没有找到不需要遍历矩阵的答案，并且一次构建张量一个单元格。

我有的解决方案：

size = len(sample["edges"])
edge_mask = torch.zeros([size, size])

for i, row in enumerate(sample["edges"]):
    for j, v in enumerate(row):
        if v != "":
            edge_mask[i, j] = 1

【问题讨论】：

标签： pytorch

【解决方案1】：

您可以将其转换为布尔数组，然后使用 torch.from_numpy，然后转换为 int：

torch.from_numpy(np.array(sample["edges"], dtype=bool)).to(int)

【讨论】：