Skip to content

P
Paddle

PaddlePaddle / Paddle
大约 2 年 前同步成功

通知 2325
Star 20933
Fork 5424
P
Paddle

add kernel for squeeze_op, test=develop !19656

!19656 已合并 由 saxon_zh@saxon_zh 创建

Created by: JepsonWong

Add kernel for squeeze_op.

  • The SqueezeOp and SqueezeGradOp class originally inherited the framework::OperatorBase class, so it cannot be used in dygraph mode.

  • By modifying the SqueezeOp and SqueezeGradOp class to inherit the framework::OperatorWithKernel class, this op can be used in the dygraph mode.