From 59e200a336e05e7458a8434fa3d0ae5547bd337f Mon Sep 17 00:00:00 2001 From: Haibin Lin Date: Tue, 4 Aug 2020 17:01:23 -0700 Subject: [PATCH] fix nn.dense doc (#18830) Co-authored-by: Lin --- python/mxnet/gluon/nn/basic_layers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/mxnet/gluon/nn/basic_layers.py b/python/mxnet/gluon/nn/basic_layers.py index 0f035eec31d3..ff6bc18551a5 100644 --- a/python/mxnet/gluon/nn/basic_layers.py +++ b/python/mxnet/gluon/nn/basic_layers.py @@ -154,7 +154,7 @@ class Dense(HybridBlock): r"""Just your regular densely-connected NN layer. `Dense` implements the operation: - `output = activation(dot(input, weight) + bias)` + `output = activation(dot(input, weight.T) + bias)` where `activation` is the element-wise activation function passed as the `activation` argument, `weight` is a weights matrix created by the layer, and `bias` is a bias vector created by the layer