TensorDesc构造函数
函数功能
构造TensorDesc。
函数原型
TensorDesc(dtype: DataType, shape: Union[List[int], Tuple[int]])
参数说明
参数名 |
数据类型 |
取值说明 |
---|---|---|
dtype |
DataType |
表示Tensor数据类型。 |
shape |
Union[List[int], Tuple[int]] |
表示Tensor的shape的描述。 |
返回值
正常场景下无返回值。
传入数据类型错误情况下会抛出TypeError异常。
调用示例
from llm_datadist import TensorDesc, DataType tensor_desc = TensorDesc(DataType.DT_FLOAT, [1,2])
约束说明
无
父主题: TensorDesc