TensorProtocol

class onnx_ir.TensorProtocol(*args, **kwargs)

Concrete tensor backed by data.

The protocol does not specify how the data is stored. That data is exposed through the raw attribute for examination, but accessing raw is typically not needed.

To use the tensor as a numpy array, call numpy(). To convert the tensor to a byte string for serialization, call tobytes().

It is recommended to check the size of the tensor first before accessing the underlying data, because accessing the data may be expensive and incur IO overhead.

name

The name of the tensor.

Type:

str | None

shape

The shape of the tensor.

Type:

ShapeProtocol

dtype

The data type of the elements of the tensor. It is an ir.DataType enum.

Type:

_enums.DataType

doc_string

Documentation string.

Type:

str | None

raw

The raw data behind this tensor. It can be anything.

Type:

Any

metadata_props

Metadata that will be serialized to the ONNX file.

Type:

MutableMapping[str, str]

meta

Metadata store for graph transform passes.

Type:

MutableMapping[str, Any]

property nbytes: int

The number of bytes required to store the tensor.

numpy()

Return the tensor as a numpy array.

Return type:

np.ndarray

property size: int

The number of elements in the tensor.

tobytes()

Return the tensor as a byte string conformed to the ONNX specification, in little endian.

Return type:

bytes