UPDATE: Read the solution here
定位问题代码：查找模型中类似 x[:， idx] = y、x[index] = y、index_put、masked_scatter、masked_fill 的用法。也可在 torch.onnx.export（...， verbose=True） 打印的导出图里查找 aten::index_put、index_put、prim::ListConstruct 等节点位置。
。体育直播对此有专业解读
Getting image source signatures
Be the first to know!