В разделе «Многослойный персептрон» в документации указано, что весовая матрица имеет размер m k. Однако I думаю, что это должно быть k m, потому что выходной слой-Y имеет размерность n * k.На странице http://mxnet.io/tutorials/python/mnist.html# неправильный размер весовой матрицы
Правильно ли я понимаю?
Правильная формула расчета матрицы для FullyConnected слоя:
output = input x weight_transpose + bias
Таким образом, в этой формуле, ваш вклад n x m, ваш вес k x m перед тем, как транспонированными, и вашим выходом n x k. См. Документ FullyConnected API doc here.
Если вы только начинаете, может возникнуть смысл сосредоточиться на новом Gluon API, который намного легче изучить и отладить.