更新时间：2021-03-18 GMT+08:00
InferShapeAndTypeFn函数
函数功能
维度推导函数。
函数原型
OpRegistrationData& InferShapeAndTypeFn(InferShapeFunc inferShapeFn);
参数说明
参数 |
输入/输出 |
说明 |
|---|---|---|
inferShapeFn |
输入 |
回调函数InferShapeFunc，请参见回调函数InferShapeFunc。 |
回调函数InferShapeFunc
用户自定义并实现InferShapeFunc类函数，用于获取算子的输出描述，包括输出shape信息、数据类型等张量描述信息。
函数原型
Status InferShapeFunc(const ge::Operator& op, vector<ge::TensorDesc>& v_output_desc);
参数说明
参数 |
输入/输出 |
说明 |
|---|---|---|
op |
输入 |
适配昇腾AI处理器的离线模型的算子数据结构。 关于Operator类，请参见《GE API参考》中的“Operator类接口”。 |
v_output_desc |
输出 |
存储算子的输出描述。 关于TensorDesc类，请参见《GE API参考》中的“TensorDesc类接口”。 |
父主题： 算子插件开发函数参考
