产品 |
是否支持 |
|---|---|
√ |
|
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。
无