I'm trying to convert Pytorch model containing NonZero operations to TRT.
I've successfully saved it as ONNX, but neither TensorRT support the operation natively nor existing converters like https://github.com/onnx/onnx-tensorrt
Could you please advice how I can deal with it maybe by changing the operation to Where or Equal + Logical Not?