提交 748e204e 编写于 作者: S sneaxiy

Revert "refine ZeroGradFunctor in activation_op.h"

This reverts commit 1eeb11ef.
上级 64292f07
...@@ -353,7 +353,7 @@ struct ZeroGradFunctor : public BaseActivationFunctor<T> { ...@@ -353,7 +353,7 @@ struct ZeroGradFunctor : public BaseActivationFunctor<T> {
template <typename Device, typename X, typename Out, typename dOut, template <typename Device, typename X, typename Out, typename dOut,
typename dX> typename dX>
void operator()(Device d, X x, Out out, dOut dout, dX dx) const { void operator()(Device d, X x, Out out, dOut dout, dX dx) const {
dx.device(d) = out.constant(static_cast<T>(0)); dx.device(d) = static_cast<T>(0) / out;
} }
}; };
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册