文章目录 完整信息 报错原因 修改方式 完整信息 /cuda/ScatterGatherKernel.cu:144: operator(): block: [0,0,0], thread: [0,0,0] Assertion `idx_dim >= 0 && idx_dim < index_size && "index out of bounds"` failed. loss_value = loss.item