Table Of Contents
Table Of Contents

mxnet.contextΒΆ

Context management API of mxnet.

cpu([device_id]) Returns a CPU context.
cpu_pinned([device_id]) Returns a CPU pinned memory context.
current_context() Returns the current context.
gpu([device_id]) Returns a GPU context.
num_gpus() Query CUDA for the number of GPUs present.
Context(device_type[, device_id]) Constructs a context.