pypto.Tensor.shape#

产品支持情况#

• Ascend 950PR：支持

• Atlas A3 训练系列产品/Atlas A3 推理系列产品：支持

• Atlas A2 训练系列产品/Atlas A2 推理系列产品：支持

功能说明#

获取Tensor Shape。

函数原型#

shape(self) -> List[SymInt]

参数说明#

无

返回值说明#

返回Tensor的形状列表。

约束说明#

无。

调用示例#

t = pypto.tensor((16, 32), pypto.DT_FP32)
out = t.shape

结果示例如下：

输出数据out: [16, 32]