文档
注册

Tensor

函数功能

构造Tensor。

函数原型

Tensor(data, tensor_desc: TensorDesc = None)

参数说明

参数名

数据类型

取值说明

data

Union[np.ndarray, Tensor]

表示Tensor的数据。

tensor_desc

TensorDesc

表示Tensor的描述信息。

调用示例

from llm_datadist import Tensor
tensor = Tensor(numpy.array([1]))

返回值

正常场景下无返回值。

传入data信息和tensor_desc信息不匹配,会抛出RuntimeError。

约束说明

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词