TensorProtocol¶

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

由数据支持的具体张量。

该协议未指定数据如何存储。数据通过 raw 属性公开以供检查，但通常不需要访问 raw。

要将张量用作 numpy 数组，请调用 numpy()。要将张量转换为字节字符串以进行序列化，请调用 tobytes()。

建议在访问底层数据之前先检查张量的大小，因为访问数据可能成本高昂并产生 IO 开销。

name¶

张量的名称。

shape¶

张量的形状。

dtype¶

张量元素的 DataTypes。它是一个 ir.DataType 枚举。

doc_string¶

文档字符串。

raw¶

此张量背后的原始数据。它可以是任何类型。

metadata_props¶

将序列化到 ONNX 文件中的元数据。

meta¶

用于图转换传递的元数据存储。

将张量作为 numpy 数组返回。

将张量作为符合 ONNX 规范的字节字符串（小端格式）返回。