mxnet随笔－ Dense层基础

from mxnet import nd
from mxnet.gluon import nn
x = nd.array(((1,2,3),(5,6,7)))
#创建神经网络第一层密集层,Dense会自动计算输入的维度，这里的5是指输出的维度的第2维的值。
layer = nn.Dense(5)
#创建其权值为均匀地抽取随机值[−0.7,0.7]
layer.initialize()
print(layer)
print(layer(x))#经过层计算前向输出

dense ：全连接层 相当于添加一个层，也称为密集层。

Dense(None -> 5, linear)

[[ 0.15825659 -0.14135249 -0.21782206 0.26669252 0.09364527]
[ 0.3746552 -0.28442633 -0.6347311 0.6351818 0.25521633]]

创建了一个线性密集层。

print(layer.weight.data())

[[ 0.05181222 0.06700657 -0.00369488]
[ 0.0418822 0.0421275 -0.00539289]
[ 0.00286685 0.03927409 0.02504314]
[-0.05344158 0.03088857 0.01958894]
[ 0.01148278 -0.04993054 0.00523225]]