onnx_ir.tensor_adapters¶

兼容的适配器，为各种框架张量类型实现 TensorProtocol 接口。

此模块提供了公共类，它们为来自流行深度学习框架的各种张量类型实现了 onnx_ir.TensorProtocol 接口。

您可以使用这些类创建张量，并在 IR 图中使用它们，就像使用任何其他张量一样。

示例：

import torch import onnx_ir as ir

# 创建一个 PyTorch 张量 torch_tensor = torch.tensor([1, 2, 3])

# 将 PyTorch 张量包装到 TorchTensor 对象中 ir_tensor = ir.tensor_adapters.TorchTensor(torch_tensor)

# 在图中使用 IR 张量 attr = ir.AttrTensor(“x”, ir_tensor) print(attr)

PyTorch 适配器¶

onnx_ir.tensor_adapters.from_torch_dtype(dtype)[source]¶

将 PyTorch dtype 转换为 ONNX IR DataType。

onnx_ir.tensor_adapters.to_torch_dtype(dtype)[source]¶

将 ONNX IR DataType 转换为 PyTorch dtype。