Browse Source

fixed LeakyReLU

tags/v0.5.0-beta
jiangjinsheng 5 years ago
parent
commit
6c92282e5e
1 changed files with 4 additions and 4 deletions
  1. +4
    -4
      mindspore/nn/layer/activation.py

+ 4
- 4
mindspore/nn/layer/activation.py View File

@@ -249,11 +249,11 @@ class LeakyReLU(Cell):
self.alpha = alpha

def construct(self, x):
alpha = P.Cast()(F.scalar_to_array(self.alpha), P.DType()(x))
if alpha <= 1:
out = P.Maximum()(alpha * x, x)
alpha_array = P.Cast()(F.scalar_to_array(self.alpha), P.DType()(x))
if self.alpha <= 1:
out = P.Maximum()(alpha_array * x, x)
else:
out = P.Minimum()(alpha * x, x)
out = P.Minimum()(alpha_array * x, x)
return out




Loading…
Cancel
Save