diff options
| author | Marcin Junczys-Dowmunt <junczys@amu.edu.pl> | 2017-09-10 14:17:17 +0300 |
|---|---|---|
| committer | Marcin Junczys-Dowmunt <junczys@amu.edu.pl> | 2017-09-10 14:17:17 +0300 |
| commit | 5cbf89b8cb2f7179b2cac14b87c2f2ef401e5e10 (patch) | |
| tree | f02f53efddd1b7e4ad06ffb7e843d7698f92ee27 /src/graph/node_operators_unary.h | |
| parent | 196879c39229dbf4ff1f212e27f80afe0ccb2e9e (diff) | |
towards a transformer-style encoder
Diffstat (limited to 'src/graph/node_operators_unary.h')
| -rw-r--r-- | src/graph/node_operators_unary.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/graph/node_operators_unary.h b/src/graph/node_operators_unary.h index 0d3981fe..3d4870e8 100644 --- a/src/graph/node_operators_unary.h +++ b/src/graph/node_operators_unary.h @@ -183,7 +183,7 @@ struct SoftmaxNodeOp : public NaryNodeOp { template <typename... Args> SoftmaxNodeOp(Expr a, Expr mask, Args... args) - : NaryNodeOp({a, mask}, args...), mask_(mask) {} + : NaryNodeOp({a}, args...), mask_(mask) {} Expr mask_; |