構(gòu)造具有靈活正向函數(shù)的網(wǎng)絡(luò)
創(chuàng)建nn.Block一個子類和實現(xiàn)兩種方法:

from mxnet import nd from mxnet.gluon import nn class MixMLP(nn.Block):def __init__(self, **kwargs):super(MixMLP, self).__init__(**kwargs)self.blk = nn.Sequential()self.blk.add(nn.Dense(5, activation='relu'),nn.Dense(7, activation='relu'))self.dense = nn.Dense(4)def forward(self, x):y = nd.relu(self.blk(x))print(y)return self.dense(y)net = MixMLP() print(net) net.initialize()x = nd.random.uniform(shape=(2,3)) y = net(x) print(y.shape) print(net.blk[0].weight.data().shape, net.blk[1].weight.data().shape)

MixMLP(
(blk): Sequential(
(0): Dense(None -> 5, Activation(relu))
(1): Dense(None -> 7, Activation(relu))
)
(dense): Dense(None -> 4, linear)
)

[[0.00156497 0.000491 0. 0. 0.0014739 0.
0.00275145]
[0.00237477 0.00133392 0. 0. 0.00215758 0.00239905
0.00092076]]
<NDArray 2x7 @cpu(0)>
(2, 4)
(5, 3) (7, 5)

總結(jié)

以上是生活随笔為你收集整理的mxnet基础到提高(7)--卷积神经网络基础(2)的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。