我尝试遵循这个不错的MXNet Tutorial。我创建了一个非常简单的神经网络(两个输入单元,没有隐藏单元和一个输出单元):
from mxnet import gluon
net = gluon.nn.Dense(1, in_units=2)
之后我尝试看一下权重矩阵的形状(与教程中描述的方式相同):
print(net.weight)
因此我希望看到这个:
Parameter dense4_weight (shape=(1, 2), dtype=None)
但是,我看到以下错误消息:
Traceback (most recent call last):
File "tmp.py", line 5, in <module>
print(net.weight)
File "/usr/local/lib/python3.6/site-packages/mxnet/gluon/parameter.py", line 120, in __repr__
return s.format(**self.__dict__)
KeyError: 'shape'
我做错了吗?