chainerx.using_device¶
-
chainerx.
using_device
(device)¶ Creates a context manager to temporarily set the default device.
- Parameters
device (Device or str) – Device object or device name to set as the default device during the context. See
chainerx.Device.name
for the specification of device names.