set_shape

产品支持情况

产品	是否支持
Atlas 350 加速卡	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	√
Atlas 推理系列产品	√
Atlas 训练系列产品	√

功能说明

设置张量形状。

函数原型

set_shape(shape: List[int]) -> TensorHolder

参数说明

参数名	输入/输出	描述
shape	输入	形状维度列表。

返回值说明

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

约束说明

如果shape不是整数列表，抛出TypeError。
如果所有形状维度不是整数，抛出TypeError。
如果操作失败，抛出RuntimeError。

父主题： TensorHolder