StringTensor

class onnx_ir.StringTensor(value, *, shape=None, name=None, doc_string=None, metadata_props=None)

字符串的多维数组(作为二进制数据以匹配 TensorProto 中的 string_data 字段)。

参数:
  • value (Sequence[bytes] | npt.NDArray[np.bytes_])

  • shape (Shape | None)

  • name (str | None)

  • doc_string (str | None)

  • metadata_props (dict[str, str] | None)

display(*, page=False)

漂亮地打印对象。

参数:

page (bool) – 是否分页输出。

返回类型:

property doc_string: str | None

文档字符串。

property dtype: DataType

张量的数据类型。不可变。

property meta: MetadataStore

用于中间分析的元数据存储。

如果您希望将元数据序列化为 ONNX proto,请写入 metadata_props

property metadata_props: dict[str, str]

张量的元数据属性。

元数据属性用于存储有关张量的附加信息。与 meta 不同,此属性被序列化为 ONNX 协议。

property name: str | None

张量的名称。

property nbytes: int

张量中的字节数。

numpy()[source]

将张量作为 numpy 数组返回。

返回类型:

npt.NDArray[np.bytes_]

property raw: Sequence[bytes] | npt.NDArray[np.bytes_]

张量的后台数据。不可变。

property shape: Shape

张量的形状。不可变。

property size: int

张量中的元素数量。

string_data()[source]

返回张量的字符串数据。

返回类型:

Sequence[bytes]

tobytes()[source]

以小端格式返回符合 ONNX 规范的字节字符串形式的张量。

返回类型:

bytes

tofile(file)

将张量写入二进制文件。

此方法将张量的原始字节写入类文件对象。类文件对象必须具有接受字节的 write 方法。

在 0.1.11 版本中添加。

参数:

file – 具有接受字节的 write 方法的类文件对象。

返回类型: