函数:set_tensor_shape
C函数原型 |
aclError aclSetTensorShape(aclTensorDesc *desc, int numDims, const int64_t *dims) |
---|---|
Python函数 |
ret = acl.set_tensor_shape(desc, dims) |
函数功能 |
调用acl.create_tensor_desc接口创建tensor描述信息后,可通过acl.set_tensor_shape设置昇腾AI处理器内部处理tensor时的dims信息,当前主要用于pytorch场景下,同步接口。
说明:
例如,某算子的原始Format为NHWC(原始Shape为4维)或者NDHWC(原始Shape为5维),为了提高数据访问效率,在进行模型转换或者在线构建网络时会自动转换Format,将原始数据Format转化为昇腾AI处理器内部计算数据时用的Format(NC1HWC0或者NDC1HWC0),同时进行推导Shape用于内部计算。 |
输入说明 |
desc:int,tensor描述数据的指针。 dims:list,要设置的dims。 |
返回值说明 |
ret:int,错误码。
|
约束说明 |
无 |
注意事项 |
无 |
父主题: 公共接口