Tensor Constructor

Applicable Products

Product	Supported (Yes/No)
Ascend 950PR/Ascend 950DT	No
Atlas A3 training products/Atlas A3 inference products	Yes
Atlas A2 training products/Atlas A2 inference products	Yes
Atlas 200I/500 A2 inference products	No
Atlas inference products	No
Atlas training products	No

Note: For the Atlas A2 training products/Atlas A2 inference products, only the Atlas 800I A2 inference server and A200I A2 Box heterogeneous subrack are supported.

Function Description

Constructs a tensor.

Prototype

__init__(data, tensor_desc: TensorDesc = None)

Parameters

Parameter	Data Type	Value Description
data	Union[np.ndarray, Tensor]	Tensor data.
tensor_desc	TensorDesc	Description of tensor.

Example

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

Returns

In normal cases, the Tensor instance is returned.

If the input data information does not match the tensor description information, RuntimeError is reported.

Constraints

None

Parent topic: Tensor