From 12aa35f43234727118e50daafd5b2287056bff41 Mon Sep 17 00:00:00 2001
From: littletomatodonkey <dazhiningsibuqu@163.com>
Date: Mon, 19 Oct 2020 17:38:13 +0000
Subject: [PATCH] fix ones

---
 ppgan/models/losses.py | 12 ++++++++----
 ppgan/modules/init.py  |  2 +-
 2 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/ppgan/models/losses.py b/ppgan/models/losses.py
index 3fea20e..6da51e2 100644
--- a/ppgan/models/losses.py
+++ b/ppgan/models/losses.py
@@ -48,13 +48,17 @@ class GANLoss(nn.Layer):
         """
         if target_is_real:
             if not hasattr(self, 'target_real_tensor'):
-                self.target_real_tensor = self.target_real_label * paddle.ones(
-                    paddle.shape(prediction), dtype='float32')
+                self.target_real_tensor = paddle.fill_constant(
+                    shape=paddle.shape(prediction),
+                    value=self.target_real_label,
+                    dtype='float32')
             target_tensor = self.target_real_tensor
         else:
             if not hasattr(self, 'target_fake_tensor'):
-                self.target_fake_tensor = self.target_fake_label * paddle.ones(
-                    paddle.shape(prediction), dtype='float32')
+                self.target_fake_tensor = paddle.fill_constant(
+                    shape=paddle.shape(prediction),
+                    value=self.target_fake_label,
+                    dtype='float32')
             target_tensor = self.target_fake_tensor
 
         # target_tensor.stop_gradient = True
diff --git a/ppgan/modules/init.py b/ppgan/modules/init.py
index 0a9fcee..6b33cc1 100644
--- a/ppgan/modules/init.py
+++ b/ppgan/modules/init.py
@@ -80,7 +80,7 @@ def calculate_gain(nonlinearity, param=None):
 
 @paddle.no_grad()
 def constant_(x, value):
-    temp_value = value * paddle.ones(x.shape, x.dtype)
+    temp_value = paddle.full(shape=x.shape, fill_value=value, dtype=x.dtype)
     x.set_value(temp_value)
     return x
 
-- 
GitLab