在 ONNX IR 中的张量表示#
ONNX IR提供了 ir.TensorProtocol
接口,以便使用不同的数据结构作为张量的后备数据。除了传统的 onnx.TensorProto
之外,您还可以使用 np.ndarray
、torch.Tensor
、jax.Array
以及几乎任何其他东西来表示计算图中的张量。这使得它们可以通过相同的 TensorProtocol
接口进行访问和序列化,而在初始化期间不会发生额外的复制。
ONNX IR提供了 ir.TensorProtocol
接口,以便使用不同的数据结构作为张量的后备数据。除了传统的 onnx.TensorProto
之外,您还可以使用 np.ndarray
、torch.Tensor
、jax.Array
以及几乎任何其他东西来表示计算图中的张量。这使得它们可以通过相同的 TensorProtocol
接口进行访问和序列化,而在初始化期间不会发生额外的复制。