函数：set_tensor_format
C函数原型 |
aclError aclSetTensorFormat(aclTensorDesc *desc, aclFormat format) |
|---|---|
Python函数 |
ret = acl.set_tensor_format(desc, format) |
函数功能 |
调用acl.create_tensor_desc接口创建Tensor描述信息后，可通过acl.set_tensor_format设置昇腾AI处理器内部处理Tensor时的Format信息，当前主要用于PyTorch场景下。 Atlas 训练系列产品，不支持该接口。
说明：
例如，某算子的原始Format为NHWC（原始Shape为4维）或者NDHWC（原始Shape为5维），为了提高数据访问效率，在进行模型转换或者在线构建网络时会自动转换Format，将原始数据Format转化为昇腾AI处理器内部计算数据时用的Format（NC1HWC0或者NDC1HWC0），同时进行推导Shape用于内部计算。 |
输入说明 |
desc：int，Tensor描述数据的指针地址。 需提前调用acl.create_tensor_desc接口创建aclTensorDesc类型。 format：int， 指定Tensor描述的格式。 |
返回值说明 |
ret：int，错误码。
|
约束说明 |
无 |
父主题： 公共接口
