函数：set_tensor_storage_format

产品	是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件	√
Atlas 训练系列产品	x
Atlas 推理系列产品	√
Atlas 200I/500 A2 推理产品	√
Atlas 200/300/500 推理产品	√

调用acl.create_tensor_desc接口创建Tensor描述信息后，可通过acl.set_tensor_storage_format设置Tensor的实际Format信息，当前主要用于PyTorch场景下。

此接口后续版本会废弃，请使用acl.set_tensor_format接口。

C函数原型

aclError aclSetTensorStorageFormat(aclTensorDesc *desc, aclFormat format)

python函数

ret = acl.set_tensor_storage_format(desc, format)

参数名	说明
desc	int，aclTensorDesc类型的指针地址。需提前调用acl.create_tensor_desc接口创建aclTensorDesc类型。
format	int，指定Tensor描述的格式。

参数名

说明

desc

int，aclTensorDesc类型的指针地址。

需提前调用acl.create_tensor_desc接口创建aclTensorDesc类型。

format

int，指定Tensor描述的格式。

返回值	说明
ret	int，错误码。返回0表示成功。返回其它值表示失败。

父主题： aclmdlTensorDesc