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