device_guard

• device (str|None) – 指定上下文中使用的设备。它可以是’cpu’或者’gpu‘，当它被设置为’cpu’或者’gpu’时，创建在该上下文中的OP将被运行在CPUPlace或者CUDAPlace上。若设置为’gpu’，同时程序运行在单卡模式下，设备的索引将与执行器的设备索引保持一致。默认值：None，在该上下文中的OP将被自动地分配设备。