當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


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。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。