Tensor构造函数
产品支持情况
产品 |
是否支持 |
|---|---|
√ |
|
√ |
|
x |
|
x |
|
x |
|
x |
函数功能
构造Tensor。
函数原型
1 | __init__(data, tensor_desc: TensorDesc = None) |
参数说明
参数名 |
数据类型 |
取值说明 |
|---|---|---|
data |
Union[np.ndarray, Tensor] |
表示Tensor的数据。 |
tensor_desc |
表示Tensor的描述信息。 |
调用示例
1 2 | from llm_datadist import Tensor tensor = Tensor(numpy.array([1])) |
返回值
正确情况下返回Tensor的实例。
传入data信息和tensor_desc信息不匹配时，会抛出RuntimeError。
约束说明
无
父主题： Tensor