变量

class onnx_graphsurgeon.Variable(name: str, dtype: dtype | onnx.TensorProto.DataType = None, shape: Sequence[int | str] = None, type: str = 'tensor_type')

基类: Tensor

表示一个张量，其值在推理时才知道。

参数:

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

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

• shape (Sequence[Union[int, str]]) – 张量的形状。如果模型使用维度参数，则可能包含字符串。

• type (str) – 张量的类型。

to_constant(values: ndarray, export_dtype: dtype | onnx.TensorProto.DataType = None)

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

参数:

• values (np.ndarray) – 此张量中的值

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

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

返回:

self

copy()

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

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