torch.onnx

API名称

是否支持

限制与说明

torch.onnx.export

是

  

torch.onnx.export_to_pretty_string

是

  

torch.onnx.register_custom_op_symbolic

是

  

torch.onnx.unregister_custom_op_symbolic

否

  

torch.onnx.select_model_mode_for_export

是

  

torch.onnx.is_in_onnx_export

是

  

torch.onnx.enable_log

否

  

torch.onnx.disable_log

否

  

torch.onnx.verification.find_mismatch

否

  

torch.onnx.JitScalarType

否

  

torch.onnx.JitScalarType.dtype

否

  

torch.onnx.JitScalarType.from_dtype

否

  

torch.onnx.JitScalarType.from_value

否

  

torch.onnx.JitScalarType.onnx_compatible

否

  

torch.onnx.JitScalarType.onnx_type

否

  

torch.onnx.JitScalarType.scalar_name

否

  

torch.onnx.JitScalarType.torch_name

否

  

torch.onnx.verification.GraphInfo

否

  

torch.onnx.verification.GraphInfo.all_mismatch_leaf_graph_info

否

  

torch.onnx.verification.GraphInfo.clear

否

  

torch.onnx.verification.GraphInfo.essential_node_count

否

  

torch.onnx.verification.GraphInfo.essential_node_kinds

否

  

torch.onnx.verification.GraphInfo.export_repro

否

  

torch.onnx.verification.GraphInfo.find_mismatch

否

  

torch.onnx.verification.GraphInfo.find_partition

否

  

torch.onnx.verification.GraphInfo.has_mismatch

否

  

torch.onnx.verification.GraphInfo.pretty_print_mismatch

否

  

torch.onnx.verification.GraphInfo.pretty_print_tree

否

  

torch.onnx.verification.GraphInfo.verify_export

否

  

torch.onnx.verification.VerificationOptions

否