TensorDesc

函数功能

构造TensorDesc。

函数原型

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

参数说明

返回值

正常场景下无返回值。

传入数据类型错误情况下会抛出TypeError异常。

调用示例

from llm_datadist import TensorDesc, DataType
tensor_desc = TensorDesc(DataType.DT_FLOAT, [1,2])

约束说明

无