SetInferShape 函数功能注册Shape推导函数。 函数原型OpDef &SetInferShape(gert::OpImplRegisterV2::InferShapeKernelFunc func); 参数说明 参数 输入/输出 说明 func 输入 Shape推导函数。InferShapeKernelFunc类型定义如下: using InferShapeKernelFunc = UINT32 (*)(InferShapeContext *); 返回值说明OpDef算子定义,OpDef请参考OpDef。 约束说明无 父主题: OpDef