用法:
mxnet.context.cpu_pinned(device_id=0)
device_id:(
int
,
optional
) - 設備的設備 ID。device_id
CPU 不需要。這包括使接口與 GPU 兼容。context:- 相應的 CPU 固定內存上下文。
參數:
返回:
返回類型:
返回 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)
相關用法
- Python mxnet.context.cpu用法及代碼示例
- Python mxnet.context.current_context用法及代碼示例
- Python mxnet.context.gpu用法及代碼示例
- Python mxnet.context.Context.device_type用法及代碼示例
- Python mxnet.context.Context用法及代碼示例
- Python mxnet.context.Context.empty_cache用法及代碼示例
- Python mxnet.contrib.ndarray.index_copy用法及代碼示例
- Python mxnet.contrib.symbol.SparseEmbedding用法及代碼示例
- Python mxnet.contrib.symbol.edge_id用法及代碼示例
- Python mxnet.contrib.symbol.dgl_graph_compact用法及代碼示例
- Python mxnet.contrib.symbol.dgl_adjacency用法及代碼示例
- Python mxnet.contrib.ndarray.hawkesll用法及代碼示例
- Python mxnet.contrib.ndarray.SparseEmbedding用法及代碼示例
- Python mxnet.contrib.ndarray.ifft用法及代碼示例
- Python mxnet.contrib.ndarray.group_adagrad_update用法及代碼示例
- Python mxnet.contrib.ndarray.fft用法及代碼示例
- Python mxnet.contrib.ndarray.boolean_mask用法及代碼示例
- Python mxnet.contrib.symbol.dgl_csr_neighbor_non_uniform_sample用法及代碼示例
- Python mxnet.contrib.ndarray.dgl_subgraph用法及代碼示例
- Python mxnet.contrib.tensorboard.LogMetricsCallback用法及代碼示例
注:本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.context.cpu_pinned。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。