Function format_tensor

dali/python/nvidia/dali/_tensor_formatting.py:242–285

Format a single tensor for display. Parameters ---------- obj : Any The tensor object to format. show_data : bool, optional Whether to include the actual data values, by default True. adapter : TensorAdapter, optional Adapter for accessing tensor properti

(obj, show_data: bool = True, adapter: Optional[TensorAdapter] = None)

def format_tensor(obj, show_data: bool = True, adapter: Optional[TensorAdapter] = None) -> str:
    """Format a single tensor for display.

    Parameters
    ----------
    obj : Any
        The tensor object to format.
    show_data : bool, optional
        Whether to include the actual data values, by default True.
    adapter : TensorAdapter, optional
        Adapter for accessing tensor properties. If None, uses PipelineTensorAdapter.

    Returns
    -------
    str
        Formatted string representation of the tensor.
    """

    if adapter is None:
        adapter = PipelineTensorAdapter()

    indent = " " * 4
    edgeitems = 2
    type_name = adapter.get_type_name(obj)
    layout = adapter.get_layout(obj)
    device = adapter.get_device(obj).lower()

    if show_data:
        data = adapter.to_numpy(obj)
        data_str = np.array2string(data, prefix=indent, edgeitems=edgeitems)
    else:
        data_str = None

    shape = tuple(adapter.get_shape(obj))

    params = (
        ([f"{data_str}"] if show_data else [])
        + [f"dtype={adapter.get_dtype(obj)}"]
        + ([f'layout="{layout}"'] if layout else [])
        + [f'device="{device}"']
        + [f"shape={shape})"]
    )

    return f"{type_name}(\n{indent}" + _join_string(params, sep=",\n" + indent)


def format_batch(