CUDA ERROR: device-side assert triggered_cu:92: operator(): block: [79,0,0], thread: [79,0,

# device-side assert triggered

operator(): block: [116,0,0], thread: [95,0,0] Assertion `index >= -sizes[i] && index < sizes[i] && "index out of bounds"` failed.
1

在网上的很多资料都说这个问题是索引有问题
从报错行一行行向上check,发现确实存在索引出错的问题
在我的代码中，需要一个广播的功能
类似于，

>> a = torch.tensor([[0,0,0],[1,1,1]])
>> b = torch.tensor([0,0,0,1,1,1])
>
>> a
tensor([[0, 0, 0],
        [1, 1, 1]])
>> b
tensor([0, 0, 0, 1, 1, 1])

>> a[b[:],:]
tensor([[0, 0, 0],
        [0, 0, 0],
        [0, 0, 0],
        [1, 1, 1],
        [1, 1, 1],
        [1, 1, 1]])
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

    y = torch.zeros(h, n).to(device)
    y_ind = g_e_ind[1, :] - m  #这里经过减法没有从0开始，导致虽然个数一样，但是内容的数字在作为ind时溢出
1
2

出错原因是m为200，g_e_ind的维度也应该是200，但这里因为前面代码的原因，为800