mxnet.ndarray.random.uniform

mxnet.ndarray.random.uniform_第1张图片
uniform表示生成一个服从均匀分布的矩阵,X~U(a,b),在这个API文档里面其实就是X~U(low,high),uniform返回值会介于low和high之间,uniform的shape参数是表示生成的矩阵的形状,比如下面的代码,输入的X是(2,8)形状,然后运行的时候给uniform指定的也是X.shape,所以返回值和X的shape形状完全一样,也是2行8列

代码是李沐的深度学习的

import d2lzh as d2l
from mxnet import autograd, gluon, init, nd
from mxnet.gluon import loss as gloss, nn

def dropout666(X, drop_prob):

    mask = nd.random.uniform(0, 1, X.shape)
    print('mask=',mask)
X = nd.arange(16).reshape((2, 8))
dropout666(X,0.4)

mxnet.ndarray.random.uniform_第2张图片

你可能感兴趣的:(计算机视觉,mxnet,人工智能)