avg_pool3d算子

原创

已于 2025-09-11 11:19:05 修改 · 107 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#目标检测 #pytorch

于 2025-04-05 21:59:24 首次发布

import numpy as np
from onnx import TensorProto,numpy_helper
from onnx.helper import (
    make_node, make_tensor_value_info, make_graph, make_model
)
# import easydict
import onnxruntime
import numpy as np

print("Supported onnxruntime version: ", onnxruntime.__version__)
print("Supported Opset versions: ", onnxruntime.get_available_providers())

class AvgPool3d():        
    
    def __init__(self,):                
        pass
        
    def inference_onnx(self,inputs,args):
        
        x = make_tensor_value_info(name="x",elem_type=TensorProto.FLOAT, shape=[1,8,