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。
约束说明
无
父主题： Tensor
