代码之家  ›  专栏  ›  技术社区  ›  Roman

为什么mxnet的权重矩阵。胶子。nn。致密物体没有形状?

  •  0
  • Roman  · 技术社区  · 7 年前

    我试着跟着这个很好 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'
    

    我做错什么了吗?

    1 回复  |  直到 7 年前
        1
  •  2
  •   Sina Afrooze    7 年前

    这是一种倒退 here 并已固定在主分支上 here . 预计它将在下一个MXNet版本中修复。