当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。