Python mxnet.context.Context用法及代码示例

用法: class mxnet.context.Context(device_type, device_id=0)

参数：

device_type：({'cpu', 'gpu'} or Context.) - 表示设备类型的字符串。
device_id：(int (default=0)) - 设备的设备 ID，GPU 需要。

基础：object

构造一个上下文。

MXNet 可以在 CPU 和不同的 GPU 上运行操作。上下文说明了应在其上进行计算的设备类型和 ID。

可以简称为mx.cpu 和mx.gpu。

属性

device_type

返回当前上下文的设备类型。

注意：

上下文也可以用作更改默认上下文的一种方式。

例子：

>>> # array on cpu
>>> cpu_array = mx.nd.ones((2, 3))
>>> # switch default context to GPU(2)
>>> with mx.Context(mx.gpu(2)):
...     gpu_array = mx.nd.ones((2, 3))
>>> gpu_array.context
gpu(2)

还可以在创建数组时显式指定上下文。

>>> gpu_array = mx.nd.ones((2, 3), mx.gpu(1))
>>> gpu_array.context
gpu(1)

相关用法

注：本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.context.Context。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。