shape

功能说明

获取Tensor形状。

函数原型

shape()

参数说明

无。

支持的芯片型号

昇腾310 AI处理器

昇腾910 AI处理器

昇腾310P AI处理器AI Core

昇腾310P AI处理器Vector Core

注意事项

无

返回值

shape形状： List类型，List中的元素为立即数（int）、Scalar（int、uint）或Expr（int、uint）。

调用示例

from tbe import tik 
tik_instance = tik.Tik() 
data_A = tik_instance.Tensor("float16", (128,), name="data_A", scope=tik.scope_ubuf) 
data_A.shape

返回：[128]