MCPcopy Create free account
hub / github.com/VincentHancoder/SSGD / get_ort_model_output

Function get_ort_model_output

tests/test_onnx/utils.py:88–115  ·  view source on GitHub ↗

Run the model in onnxruntime env. Args: feat (list[Tensor]): A list of tensors from torch.rand, each is a 4D-tensor. Returns: list[np.array]: onnxruntime infer result, each is a np.array

(feat, onnx_io='tmp.onnx')

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

ort_validateFunction · 0.85

Calls 1

numpyMethod · 0.80

Tested by

no test coverage detected