aclSetTensorShape
aclSetTensorStorageFormat
须知
此接口后续版本会废弃,请使用aclSetTensorFormat接口。
函数功能
调用aclCreateTensorDesc接口创建tensor描述信息后,可通过aclSetTensorStorageFormat设置tensor的实际Format信息,当前主要用于pytorch场景下。同步接口。
函数原型
aclError aclSetTensorStorageFormat(aclTensorDesc *desc, aclFormat format)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| desc | 输出 | aclTensorDesc类型的指针。 需提前调用aclCreateTensorDesc接口创建aclTensorDesc类型。 |
| format | 输入 | 需要设置的format。 |
返回值说明
返回0表示成功,返回其它值表示失败。
父主题: aclTensorDesc
aclSetTensorOriginShape
函数功能
调用aclCreateTensorDesc接口创建tensor描述信息后,可通过aclSetTensorOriginShape接口设置tensor的原始dims信息,当前主要用于pytorch场景下。同步接口。
:::note 说明 例如,某算子的原始Format为NHWC(原始Shape为4维)或者NDHWC(原始Shape为5维),为了提高数据访问效率,在进行模型转换或者在线构建网络时会自动转换Format,将原始数据Format转化为昇腾AI处理器内部计算数据时用的Format(NC1HWC0或者NDC1HWC0),同时进行推导Shape用于内部计算。 :::
函数原型
aclError aclSetTensorOriginShape(aclTensorDesc *desc, int numDims, const int64_t *dims)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| desc | 输出 | aclTensorDesc类型的指针。 需提前调用aclCreateTensorDesc接口创建aclTensorDesc类型。 |
| numDims | 输入 | 需要设置的dims维度数。 |
| dims | 输入 | dims的指针。 |
返回值说明
返回0表示成功,返回其它值表示失败。
父主题: aclTensorDesc
aclSetTensorOriginFormat
函数功能
调用aclCreateTensorDesc接口创建tensor描述信息后,可通过aclSetTensorOriginFormat设置tensor的原始Format信息,当前主要用于pytorch场景下。同步接口。
:::note 说明 例如,某算子的原始Format为NHWC(原始Shape为4维)或者NDHWC(原始Shape为5维),为了提高数据访问效率,在进行模型转换或者在线构建网络时会自动转换Format,将原始数据Format转化为昇腾AI处理器内部计算数据时用的Format(NC1HWC0或者NDC1HWC0),同时进行推导Shape用于内部计算。 :::
函数原型
aclError aclSetTensorOriginFormat(aclTensorDesc *desc, aclFormat format)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| desc | 输出 | aclTensorDesc类型的指针。 需提前调用aclCreateTensorDesc接口创建aclTensorDesc类型。 |
| format | 输入 | 需要设置的format。 |
返回值说明
返回0表示成功,返回其它值表示失败。
父主题: aclTensorDesc
aclSetTensorShape
函数功能
调用aclCreateTensorDesc接口创建tensor描述信息后,可通过aclSetTensorShape接口设置昇腾AI处理器内部处理tensor时的dims信息,当前主要用于pytorch场景下。同步接口。
:::note 说明 例如,某算子的原始Format为NHWC(原始Shape为4维)或者NDHWC(原始Shape为5维),为了提高数据访问效率,在进行模型转换或者在线构建网络时会自动转换Format,将原始数据Format转化为昇腾AI处理器内部计算数据时用的Format(NC1HWC0或者NDC1HWC0),同时进行推导Shape用于内部计算。 :::
函数原型
aclError aclSetTensorShape(****aclTensorDesc *desc, int numDims, const int64_t *dims)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| desc | 输出 | aclTensorDesc类型的指针。 需提前调用aclCreateTensorDesc接口创建aclTensorDesc类型。 |
| numDims | 输入 | 需要设置的dims维度数。 |
| dims | 输入 | dims的指针。 |
返回值说明
返回0表示成功,返回其它值表示失败。
父主题: aclTensorDesc
aclSetTensorFormat
函数功能
调用aclCreateTensorDesc接口创建tensor描述信息后,可通过aclSetTensorFormat设置昇腾AI处理器内部处理tensor时的Format信息,当前主要用于pytorch场景下。同步接口。
:::note 说明 例如,某算子的原始Format为NHWC(原始Shape为4维)或者NDHWC(原始Shape为5维),为了提高数据访问效率,在进行模型转换或者在线构建网络时会自动转换Format,将原始数据Format转化为昇腾AI处理器内部计算数据时用的Format(NC1HWC0或者NDC1HWC0),同时进行推导Shape用于内部计算。 :::
函数原型
aclError aclSetTensorFormat(****aclTensorDesc *desc, aclFormat format)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| desc | 输出 | aclTensorDesc类型的指针。 需提前调用aclCreateTensorDesc接口创建aclTensorDesc类型。 |
| format | 输入 | 需要设置的format。 |
返回值说明
返回0表示成功,返回其它值表示失败。
父主题: aclTensorDesc
aclSetTensorShapeRange
函数功能
调用aclCreateTensorDesc接口创建tensor描述信息后,可通过本接口设置tensor的各个维度的取值范围。同步接口。
使用场景:动态Shape的算子,其输入Shape中变化维度用-1表示,但每个变化维度的范围是不一样的,需要显式设置,如Shape为[16,-1,20,-1],对应的shape range可以是[[16,16],[1,128],[20,20],[1,10]],表示第一维的Shape范围固定为16,第二维的Shape范围为1到128,第三维的Shape范围固定为20,第四维的Shape范围为1到10。
函数原型
aclError aclSetTensorShapeRange(aclTensorDesc* desc, size_t dimsCount, int64_t dimsRange[][ACL_TENSOR_SHAPE_RANGE_NUM]****)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| desc | 输出 | aclTensorDesc类型的指针。 需提前调用aclCreateTensorDesc接口创建aclTensorDesc类型。 |
| dimsCount | 输入 | tensor中dims的维度个数。 |
| dimsRange | 输入 | dimsRange为每个维度的取值范围,用二维数组表示范围。 如果Shape中的维度值不是-1时,表示固定维度,该维度对应的shape range中最大值和最小值相同;否则,表示动态维度,数组的两个维度分别表示对应tensor dims中的最小值和最大值。 text<br>#define ACL_TENSOR_SHAPE_RANGE_NUM 2<br> |
返回值说明
返回0表示成功,返回其它值表示失败。
父主题: aclTensorDesc
aclSetTensorDynamicInput
函数功能
调用aclCreateTensorDesc接口创建tensor描述信息后,可通过aclSetTensorDynamicInput接口设置多个动态输入的分组标识,例如某个算子包括动态输入为x、必选输入y,其中动态输入有多个,调用aclSetTensorDynamicInput将dynamicInputName设置为x,AscendCL内部根据dynamicInputName索引到对应的多个动态输入,达到分组的目的**。**同步接口。
函数原型
aclError aclSetTensorDynamicInput(aclTensorDesc *desc, const char *dynamicInputName)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| desc | 输出 | aclTensorDesc类型的指针。 需提前调用aclCreateTensorDesc接口创建aclTensorDesc类型。 |
| dynamicInputName | 输入 | desc中动态输入分组标识的指针。 |
返回值说明
返回0表示成功,返回其它值表示失败。
父主题: aclTensorDesc
在线提单