pytorch_pfn_extras.onnx.unstrip_tensor.unstrip

pytorch_pfn_extras.onnx.unstrip_tensor.unstrip(path, out_path='')

Unstrip ONNX models and test data(.pb).

Add tensor(raw data) to the target ONNXs (and test data). Values are random following mean and variance written in meta information.

Parameters:

• path (str) – The target directory path, ONNX file, or Tensor (Protobuf) file path.

• out_path (str) – Output path to be written.

Return type:

None