导出onnx模型
import torch
import numpy as np
import pointopsclass CustomModel(torch.nn.Module):def __init__(self):super(CustomModel, self).__init__()def forward(self, nsample, xyz, new_xyz, offset, new_offset):tmp pointops.knnquery(nsample, xyz, new_xyz,…