Function: set_tensor_storage_shape

Applicability

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

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)

Parameters

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.

Returns

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

Parent topic: aclmdlTensorDesc