Function: set_tensor_storage_shape

Applicability

Product	Supported (√/x)
Atlas A3 training products/Atlas A3 inference products	√
Atlas A2 training products/Atlas A2 inference products	√
Atlas training products	x
Atlas inference products	√
Atlas 200I/500 A2 inference products	√

Function Usage

Sets the actual dims of the tensor after creating the tensor description by calling acl.create_tensor_desc. This API is mainly used in the PyTorch scenarios.

This API will be deprecated in future releases. Use acl.set_tensor_shape instead.

Prototype

C Prototype

aclError aclSetTensorStorageShape(aclTensorDesc *desc, int numDims, const int64_t *dims)

Python Function

ret = acl.set_tensor_storage_shape(desc, dims)

Parameter Description

Parameter	Description
desc	Int, pointer address of the aclTensorDesc type. Call acl.create_tensor_desc to create data of the aclTensorDesc type in advance.
dims	List, dimensions to be set.

Parameter

Description

desc

Int, pointer address of the aclTensorDesc type.

Call acl.create_tensor_desc to create data of the aclTensorDesc type in advance.

dims

List, dimensions to be set.

Return Value Description

Return Value	Description
ret	Int, error code. 0: success. Other values: failure.

Parent topic: aclmdlTensorDesc