AllocTensor

Product	Supported
Atlas A3 training product/Atlas A3 inference product	√
Atlas A2 training product/Atlas A2 inference product	√
Atlas 200I/500 A2 inference product	x
Atlas inference product	x
Atlas training product	x

Allocates tensors based on the shape, data type, and alignment size.

std::shared_ptr<Tensor> AllocTensor(const std::vector<int64_t> &shape, DataType data_type, uint32_t align = 512U)

Parameter	Input/Output	Description
shape	Input	Shape of a tensor.
data_type	Input	Data type of a tensor.
align	Input	Alignment size for the allocated memory address. The value range is [32, 64, 128, 256, 512, 1024]. This parameter is reserved.

Pointer to the allocated tensor.

NULL is returned if the tensor allocation fails.

None

Parent topic: FlowBufferFactory Data Type