dataflow.TensorDesc
函数功能
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])
约束说明
无
父主题: DataFlow构图接口