I was trying to convert pytorch model to onnx and then to tensorrt. ... find the implementation for the Scatter_Add extension (the file. view(batch, K, 1).
確定! 回上一頁