MXNet与Gluon构建网络的主要区别
发布时间
阅读量：
阅读量
学习资料可参考李沐所著的gluon讲义。
mxnet与gluon在构建网络方面的差异主要体现在四个层面。以下将以简单的drop out功能为例，逐一进行对比分析。
0 构建dropout
ndarray:
def dropout(X, drop_prob):
assert 0 <= drop_prob <= 1
keep_prob = 1 - drop_prob
# 这种情况下把全部元素都丢弃。
if keep_prob == 0:
return X.zeros_like()
mask = nd.random.uniform(0, 1, X.shape) < keep_prob
return mask * X / keep_prob
gluon:
from mxnet import nn
drop_prob = 0.2
nn.Dropout(drop_prob)
1 构建网络
ndarray:
输入维度需要进行设定（即num_
全部评论 (0)
还没有任何评论哟~