更新时间:2021-03-18 GMT+08:00
分享

GET_DYNAMIC_INPUT_SHAPE

函数功能

获取算子指定动态输入的shape,用于实现InferShape函数。

函数原型

GET_DYNAMIC_INPUT_SHAPE(op, name, index)

约束说明

无。

参数说明

参数名

输入/输出

描述

op

输入

算子对象。

name

输入

算子输入。

index

输入

指定动态输入的第index个输入。

返回值

shape值。

调用示例

提供公共函数宏封装供算子开发InferShape函数。该函数获取动态输入的shape
分享:

    相关文档

    相关产品

close