在 ONNX IR 中的张量表示

在 ONNX IR 中的张量表示#

参考:ir.TensorProtocol

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