文档
注册

函数: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描述地址对象。

format:int, 指定tensor描述的格式。

返回值说明

ret:int,错误码。

  • 返回0表示成功。
  • 返回其它值表示失败。

约束说明

注意事项

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词