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


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