From a52f8e4abc79c5332ab31933065088aa970e06df Mon Sep 17 00:00:00 2001
From: LielinJiang <50691816+LielinJiang@users.noreply.github.com>
Date: Fri, 13 May 2022 16:04:47 +0800
Subject: [PATCH] Fix decode_jpeg example code (#42733)

* fix decode_jpeg example code

* fix decode_jpeg example code
---
 python/paddle/vision/ops.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/python/paddle/vision/ops.py b/python/paddle/vision/ops.py
index e4dd4c797f..d45c652885 100644
--- a/python/paddle/vision/ops.py
+++ b/python/paddle/vision/ops.py
@@ -895,7 +895,10 @@ def decode_jpeg(x, mode='unchanged', name=None):
 
     Examples:
         .. code-block:: python
+
+            # required: gpu
             import cv2
+            import numpy as np
             import paddle
 
             fake_img = (np.random.random(
-- 
GitLab