常量

class onnx_graphsurgeon.Constant(name: str, values: ndarray | LazyValues, data_location: int = None, export_dtype: dtype | onnx.TensorProto.DataType = None)

基类: Tensor

表示值已知的张量。

参数:

• name (str) – 张量的名称。

• values (numpy.ndarray) – 此张量中的值，以 NumPy 数组的形式。

• data_location (int) – 一个枚举值，指示张量数据存储的位置。通常，这将来自 onnx.TensorProto.DataLocation。

• export_dtype (Union[np.dtype, onnx.TensorProto.DataType]) – 当导出到 onnx 时张量的数据类型。如果未指定，则将使用 values 的数据类型。

to_variable(dtype: dtype | None = None, shape: Sequence[int | str] = [])

修改此张量以就地将其转换为变量。这意味着张量的所有消费者/生产者都将看到更新。

参数:

• dtype (Union[numpy.dtype, onnx.TensorProto.DataType]) – 张量的数据类型。

• shape (Sequence[int]) – 张量的形状。

返回值:

self

copy()

对此张量进行浅拷贝，省略输入和输出信息。

注意：通常，您应该只复制图。