在页面http://mxnet.io/tutorials/python/mnist.html#上，权重矩阵维度不正确

Question

在Multilayer Perceptron小节下，文档说重量矩阵的维数为m k。但是，我认为它应该是k m，因为输出层-Y的维数为n * k

我的理解是否正确？

Answer 1

FullyConnected图层的正确矩阵计算公式为：

output = input x weight_transpose + bias

因此，在此公式中，您的输入为n x m，转换前您的权重为k x m，输出为n x k。请参阅FullyConnected API文档here。

如果您刚开始使用，那么关注新的Gluon API可能是有意义的，这会更容易学习和调试。