mxnet基础到提高(36)-单层神经网络

最新推荐文章于 2024-02-01 15:00:57 发布

「已注销」

最新推荐文章于 2024-02-01 15:00:57 发布

阅读量176

点赞数

文章标签： ai

本文链接：https://blog.csdn.net/AI_LX/article/details/102210967

版权

构造一个4个输出的单层网络，

import mxnet.ndarray as nd
from mxnet.gluon import nn

layer1=nn.Dense(4)#权重默认为平均分布的[-0.7,0.7]，4是指输出的维度的第2维的值
print(layer1)

Dense(None -> 4, linear)

前向计算

from mxnet import nd
from mxnet.gluon import nn

layer1=nn.Dense(1)
x=nd.array([[1,1],[1,-1],[-1,1],[-1,-1]])
layer1.initialize()
y=layer1(x)
print(y)

[[-0.08561561]
[-0.04872324]
[ 0.04872324]
[ 0.08561561]]
<NDArray 4x1 @cpu(0)>

nn.Dense和nn.Sequential属于nn.Block的子类

from mxnet import nd
from mxnet.gluon import nn

net=nn.Sequential()
net.add(nn.Dense(2,activation="relu"),
        nn.Dense(1,activation="relu"))
net.initialize()
y=net(x)
print(y)

[[-0.08561561]
[-0.04872324]
[ 0.04872324]
[ 0.08561561]]
<NDArray 4x1 @cpu(0)>