alloc_tensor_msg

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 FlowMsg of the tensor type based on the shape, data type, and alignment size.

alloc_tensor_msg(self, shape: Union[List[int], Tuple[int]], dtype, align:Optional[int] = 64) -> FlowMsg

Parameter	Input/Output	Description
shape	Input	Shape of a tensor.
dtype	Input	Data type of a tensor.
align	Input	Size of the allocated memory address alignment. Values: [32, 64, 128, 256, 512, 1024]. Default value: 64.

Instance of FlowMsg

If no tensor pointer can be allocated, None is returned.

None

Parent topic: Class MetaRunContext