函数:set_tensor_format
产品支持情况
产品  | 
是否支持  | 
|---|---|
√  | 
|
√  | 
|
x  | 
|
√  | 
|
√  | 
|
√  | 
使用说明
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场景下。  说明:  
例如,某算子的原始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,错误码。 
  | 
约束说明  | 
无  | 
父主题: aclmdlTensorDesc