Your browser does not seem to support JavaScript. As a result, your viewing experience will be diminished, and you have been placed in read-only mode.
Please download a browser that supports JavaScript, or enable it if it's disabled (i.e. NoScript).
Gather export onnx problem
Hi.
idxs = torch.argmax(prob, dim=-1) scores = torch.gather(prob, -1, idxs.unsqueeze(-1))
The model exported out onnx using GatherElements, is there a way to make it export in Gather rather than GatherElements?
与 神力AI社区 的连接断开,我们正在尝试重连,请耐心等待