2024年7月28日
pytorch中自定义onnx新算子并导出为onnx
摘要： import torch from torch.autograd import Function import torch.onnx # Step 1: Define custom PyTorch operator class MyCustomOp(Function): @staticmethod 阅读全文
posted @ 2024-07-28 17:46 海_纳百川 阅读(527) 评论(0) 推荐(0)
本站总访问量次