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