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


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


用法:

mxnet.context.cpu(device_id=0)

参数

device_id(int, optional) - 设备的设备 ID。device_idCPU 不需要。这包括使接口与 GPU 兼容。

返回

context- 对应的 CPU 上下文。

返回类型

mxnet.context.Context

返回 CPU 上下文。

此函数是 Context('cpu', device_id) 的快捷方式。对于大多数操作,当未指定上下文时,默认上下文为 cpu()

例子

>>> with mx.cpu():
...     cpu_array = mx.nd.ones((2, 3))
>>> cpu_array.context
cpu(0)
>>> cpu_array = mx.nd.ones((2, 3), ctx=mx.cpu())
>>> cpu_array.context
cpu(0)

相关用法


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