TorchTensor

class onnx_ir.tensor_adapters.TorchTensor(tensor, name=None, doc_string=None)

参数:

• tensor (torch.Tensor)

• name (str | None)

• doc_string (str | None)

numpy()

将张量作为 numpy 数组返回。

当数据类型不受 numpy 支持时，使用 ml_dtype 包中的数据类型。可以使用 .view() 方法将值重新解释为位表示。

返回类型:

ndarray[tuple[int, …], dtype[_ScalarType_co]]

tobytes()

以小端编码的字节形式返回该值。

当原始值不是 numpy 数组时，覆盖此方法以实现更高效的序列化。

返回类型:

bytes

tofile(file)

将张量写入二进制文件。

在 0.1.11 版本中添加。

参数:

file – 一个具有接受字节的 write 方法或具有 fileno() 方法的类文件对象。

返回类型:

无