hello.onnx.export module¶

class hello.onnx.export.Net(*args: Any, **kwargs: Any)[source]¶

Bases: Module

forward(x)[source]¶

hello.onnx.export.export_onnx(model, x, f, opset=11, device='cpu', simplify=False, rtol=0.001, atol=1e-05)[source]¶

hello.onnx.export.select_device(device='')[source]¶

hello.onnx.export.to_numpy(tensor)[source]¶