受篇幅所限,本书无法对所有用到的MXNet函数和类一一详细介绍。我们可以主动查找相关文档来做更深入的了解。
当我们想知道一个模块(module)里面提供了哪些可以调用的函数和类的时候,可以使用dir
函数。下面我们打印nd.random
模块中所有的成员或属性。
from mxnet import nd
print(dir(nd.random))
通常我们可以忽略掉由__
开头和结尾的函数(Python的特别对象)或者由单_
开头的函数(一般为内部函数)。通过其余成员的名字我们大致猜测出这个模块提供了各种随机数的生成方法,包括从均匀分布采样(uniform
)、从正态分布采样(normal
)、从泊松分布采样(poisson
)等。
当我们想了解某个函数或者类的具体用法时,可以使用help
函数。让我们以NDArray中的ones_like
函数为例,查阅它的用法。
help(nd.ones_like)
从文档信息我们了解到,ones_like
函数会创建和输入NDArray形状相同且元素为1的新的NDArray。我们可以验证一下:
x = nd.array([[0, 0, 0], [2, 2, 2]])
y = x.ones_like()
y
在Jupyter笔记本里,我们可以使用?
来将文档显示在另外一个窗口中。例如nd.ones_like?
将得到与help(nd.ones_like)
几乎一样的内容,但会显示在额外窗口里。此外,如果使用两个nd.ones_like??
,那么会额外显示该函数实现的代码。
我们也可以在MXNet的网站上查阅相关文档。访问MXNet网站 http://mxnet.apache.org/ (如图2.1所示),点击网页顶部的下拉菜单“API”可查阅各个前端语言的接口。此外,我们也可以在网页右上方含“Search”字样的搜索框中直接搜索函数或类名称。
图2.2展示了MXNet网站上有关ones_like
函数的文档。
- 每当遇到不熟悉的MXNet API时,我们可以主动查阅它的相关文档。
- 查阅MXNet文档可以使用
dir
和help
函数,或访问MXNet官网。
- 查阅NDArray支持的其他操作。