如何根据随机索引数组将向量中的元素插入到矩阵中

Question

基本上，我试图根据随机索引将向量中的元素插入到矩阵中size = 100000answer_count = 4num_range = int(1e4)a = torch.randint(-num_range, num_range, size=(size, ))b = torch.randint(-num_range, num_range, size=(size, ))answers = torch.randint(-num_range, num_range, size=(size, answer_count))for i in range(size): answers[i, np.random.randint(answer_count)] = a[i] + b[i]我试过类似的东西c = a + bpos = torch.randint(answer_count, size=(size, ))answers[:, pos] = c但我肯定做错了什么

Answer 1

我认为您需要像这样更改最后一行：

answers[np.arange(size), pos] = c

问题在于高级索引的不正确使用。要了解这些索引的区别，请尝试打印出answers[:, pos]vs. answers[np.arange(size), pos]，您会明白为什么前一个不起作用。用一行answers[np.arange(size), pos]选择每一行，同时用每一行选择所有行。