set_origin_shape

产品支持情况

全量芯片支持。

功能说明

向TensorDesc中设置Tensor的原始Shape。 该Shape是指原始网络模型的Shape。

函数原型

1

set_origin_shape(shape: List[int]) -> TensorDesc

参数说明

返回值说明

(TensorDesc) 返回自身以支持链式调用。

约束说明

如果shape不是整数列表，抛出TypeError。