Python export_to_pretty_stringの例

コード例 #1

ファイル: __init__.py プロジェクト: yinghai/pytorch

def export_to_pretty_string(*args, **kwargs) -> str:
    r"""
    Similar to :func:`export`, but returns a text representation of the ONNX
    model. Only differences in args listed below. All other args are the same
    as :func:`export`.

    Args:
      add_node_names (bool, default True): Whether or not to set
          NodeProto.name. This makes no difference unless
          ``google_printer=True``.
      google_printer (bool, default False): If False, will return a custom,
          compact representation of the model. If True will return the
          protobuf's `Message::DebugString()`, which is more verbose.

    Returns:
      A UTF-8 str containing a human-readable representation of the ONNX model.
    """
    from torch.onnx import utils
    return utils.export_to_pretty_string(*args, **kwargs)

コード例 #2

ファイル: __init__.py プロジェクト: zsk423200/pytorch

def export_to_pretty_string(*args, **kwargs):
    from torch.onnx import utils
    return utils.export_to_pretty_string(*args, **kwargs)

コード例 #3

def export_to_pretty_string(*args, **kwargs):
    r"""
    Same as :func:`export`, but returns a text representation of the exported model.
    """
    from torch.onnx import utils
    return utils.export_to_pretty_string(*args, **kwargs)

コード例 #4

ファイル: __init__.py プロジェクト: gtgalone/pytorch

def export_to_pretty_string(*args, **kwargs):
    from torch.onnx import utils
    return utils.export_to_pretty_string(*args, **kwargs)