Skip to content

P
Paddle

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

通知 2325
Star 20933
Fork 5424
P
Paddle

add kernel for unstack_op, test=develop !19538

!19538 已合并 由 saxon_zh@saxon_zh 创建

Created by: JepsonWong

Add kernel for unstack_op.

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

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