set_device¶
Paddle 支持包括 CPU 和 GPU 在内的多种设备运行，设备可以通过字符串标识符表示，此功能可以指定 OP 运行的全局设备。
参数¶
device (str)- 此参数确定特定的运行设备，它可以是
cpu、gpu、xpu、mlu、npu、gpu:x、xpu:x、mlu:x或者是npu:x。其中，x是 GPU、 XPU、 MLU 或者是 NPU 的编号。当device是cpu的时候，程序在 CPU 上运行，当 device 是gpu:x的时候，程序在 GPU 上运行，当 device 是mlu:x的时候，程序在 MLU 上运行，当 device 是npu:x的时候，程序在 NPU 上运行。
返回¶
Place，设置的 Place。
代码示例¶
import paddle
paddle.device.set_device("cpu")
x1 = paddle.ones(name='x1', shape=[1, 2], dtype='int32')
x2 = paddle.zeros(name='x2', shape=[1, 2], dtype='int32')
data = paddle.stack([x1,x2], axis=1)