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

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


用法:

mxnet.context.cpu_pinned(device_id=0)

参数：

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

返回：

context：- 相应的 CPU 固定内存上下文。

返回类型：

mxnet.context.Context

返回 CPU 固定内存上下文。从 CPU 固定内存复制到 GPU 比从普通 CPU 内存复制要快。

此函数是 Context('cpu_pinned', device_id) 的快捷方式。

例子：

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

相关用法


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