函数:infer_shape
C函数原型 |
aclError aclopInferShape(const char *opType, int numInputs, aclTensorDesc *inputDesc[], aclDataBuffer *inputs[], int numOutputs, aclTensorDesc *outputDesc[], aclopAttr *attr) |
---|---|
Python函数 |
ret = acl.op.infer_shape(op_type, in_desc_list, in_list, num_outputs, out_desc_list, attr) |
函数功能 |
根据算子的输入Shape、输入值推导出算子的输出Shape。 |
输入说明 |
op_type:int,指定算子类型名称。 in_desc_list:list,算子输入tensor的描述。需提前调用create_tensor_desc接口创建aclTensorDesc类型 in_list:list,算子输入tensor。 此处算子输入tensor数据的内存必须根据应用运行模式来确定,应用运行在Host时,此处需申请Host上的内存。应用运行在Device时,此处需申请Device上的内存。内存申请接口请参见内存管理。此处需申请Device上的内存。内存申请接口请参见内存管理。 num_outputs:int,算子输出tensor的数量。 attr:int,算子的属性地址对象。 |
输出说明 |
out_desc_list:list,算子输出tensor的描述。需提前调用create_tensor_desc接口创建aclTensorDesc类型。out_desc_list数组中的元素个数必须与num_outputs参数值保持一致。 |
返回值说明 |
ret:int,错误码。
|
约束说明 |
|