我试着跟着这个很好
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'
我做错什么了吗?