麻花

官方文档pyTorch版本AlexNet模型转换到caffe2版本的模型 AlexNet模型已经训练好，来自与torchvision包，现在需要转换成onnx格式的模型，转换后的结果为alexnet.onnx：import torchimport torchvisiondummy_input = torch.randn(10, 3, 224, 224, device='cud...

添加自定义的onnx operations 在pyTorch中定义的网络图以及其运算，在onnx中不一定支持，因此，需要自定义的添加operators。如果onnx支持则可以直接使用，一下是支持的网络以及运算：add (nonzero alpha not supported)sub (nonzero alpha not supported)muldivcatmmaddmm...