From 3c123172885cdccd14649b595af0ec3b081c70c6 Mon Sep 17 00:00:00 2001
From: liuxiao93 <liuxiao93@huawei.com>
Date: Mon, 7 Dec 2020 11:09:43 +0800
Subject: [PATCH] Fix bug about grad of P.LinSpace.

---
 mindspore/ops/_grad/grad_math_ops.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/mindspore/ops/_grad/grad_math_ops.py b/mindspore/ops/_grad/grad_math_ops.py
index d299babd39..90dd539b87 100755
--- a/mindspore/ops/_grad/grad_math_ops.py
+++ b/mindspore/ops/_grad/grad_math_ops.py
@@ -1191,7 +1191,7 @@ def get_bprop_inv(self):
 def get_bprop_lin_space(self):
     """Grad definition for `LinSpace` operation."""
 
-    def bprop(assist, start, stop, num, out, dout):
-        return zeros_like(assist), zeros_like(start), zeros_like(stop), zeros_like(num)
+    def bprop(start, stop, num, out, dout):
+        return zeros_like(start), zeros_like(stop), zeros_like(num)
 
     return bprop