get_shape

产品支持情况

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

功能说明

获取张量形状。

函数原型

get_shape() -> List[int]

参数说明

无

返回值说明

(List[int]) 形状列表。

约束说明

如果获取失败，抛出RuntimeError。

父主题： Tensor