dataflow.TensorDesc

产品支持情况

产品	是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√

函数功能

Tensor的描述函数。

函数原型

TensorDesc(dtype: DType, shape: Union[List[int], Tuple[int]])

参数说明

参数名称	数据类型	取值说明
dtype	DType	表示Tensor的数据类型的描述。
shape	Union[List[int], Tuple[int]]	表示Tensor的shape的描述，shape中的每个元素需要>=0的数值。

返回值

正常场景下返回None。

异常情况下会抛出DfException异常。可以通过捕捉异常获取DfException中的error_code与message查看具体的错误码及错误信息。详细信息请参考DataFlow错误码。

调用示例

tensor_desc = TensorDesc(DT_FLOAT, [1,2])

约束说明

无