diff options
| author | Marcin Junczys-Dowmunt <junczys@amu.edu.pl> | 2018-06-27 23:04:44 +0300 |
|---|---|---|
| committer | Marcin Junczys-Dowmunt <junczys@amu.edu.pl> | 2018-06-27 23:04:44 +0300 |
| commit | c6350c666f293c64f3745d20d9cb9796eed849c8 (patch) | |
| tree | d39a93fdcf734cc7abcb884992cab29702af998f /src/graph/node_operators_unary.h | |
| parent | d1d0df98d2b0df859d06fb38ba376a5a717c3bfc (diff) | |
fix transpose operator
Diffstat (limited to 'src/graph/node_operators_unary.h')
| -rw-r--r-- | src/graph/node_operators_unary.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/graph/node_operators_unary.h b/src/graph/node_operators_unary.h index bb0b66f4..0fc17d28 100644 --- a/src/graph/node_operators_unary.h +++ b/src/graph/node_operators_unary.h @@ -806,11 +806,11 @@ struct TransposeNodeOp : public UnaryNodeOp { : UnaryNodeOp(a, newShape(a, axes)), axes_{axes} {} NodeOps forwardOps() { - return {NodeOp(TransposeND(val_, child(0)->val(), axes_))}; + return {NodeOp(TransposeND(val_, child(0)->val(), axes_, 0.f))}; } NodeOps backwardOps() { - return {NodeOp(TransposeND(child(0)->grad(), adj_, axes_))}; + return {NodeOp(TransposeND(child(0)->grad(), adj_, axes_, 1.f))}; } template <class... Args> |