add paddle.device.device_guard API (#7361)

Author: HydrogenSulfate

docs/api/paddle/device/Overview_cn.rst

@@ -85,6 +85,7 @@ Stream 与 Event 相关 API
     " :ref:`current_stream <cn_api_paddle_device_current_stream>` ", "通过 device 返回当前的 stream"
     " :ref:`set_stream <cn_api_paddle_device_set_stream>` ", "设置当前的 stream"
     " :ref:`stream_guard <cn_api_paddle_device_stream_guard>` ", "切换当前的 stream 为输入指定的 stream，该 API 目前仅支持动态图模式"
+    " :ref:`device_guard <cn_api_paddle_device_device_guard>` ", "切换当前的 device 为输入指定的 device，该 API 目前仅支持动态图模式"
     " :ref:`synchronize <cn_api_paddle_device_synchronize>` ", "等待给定的设备上的计算完成"
 
 

docs/api/paddle/device/device_guard_cn.rst

@@ -0,0 +1,20 @@
+.. _cn_api_paddle_device_device_guard:
+
+device_guard
+-------------------------------
+
+.. py:function:: paddle.device.device_guard(device)
+
+可以切换当前的 device 为输入指定的 device。
+
+.. note::
+    该 API 目前仅支持动态图模式。
+
+参数
+::::::::::::
+
+    - **device** (PlaceLike) - 指定的 device，可以是形如 "cpu", "gpu:0" 之类的设备描述字符串，也可以是 `paddle.CUDAPlace(0)`, `paddle.CPUPlace()` 之类的设备实例。
+
+代码示例
+::::::::::::
+COPY-FROM: paddle.device.device_guard