比较与tf.nn.leaky_relu的差异

tf.nn.leaky_relu

tf.nn.leaky_relu(features, alpha=0.2, name=None) -> Tensor

更多内容详见tf.nn.leaky_relu。

mindspore.nn.LeakyReLU

class mindspore.nn.LeakyReLU(alpha=0.2)(x) -> Tensor

更多内容详见mindspore.nn.LeakyReLU。

差异对比

TensorFlow：应用Leaky ReLU激活函数，其中参数alpha是用于控制激活函数的斜率。

MindSpore：MindSpore此API实现功能与TensorFlow基本一致。

分类	子类	TensorFlow	MindSpore	差异
参数	参数1	features	x	功能一致，参数名不同
	参数2	alpha	alpha	-
	参数3	name	-	不涉及

代码示例

两API实现功能一致，用法相同。

# TensorFlow
import tensorflow as tf

features = tf.constant([[-1.0, 4.0, -8.0], [2.0, -5.0, 9.0]], dtype=tf.float32)
output = tf.nn.leaky_relu(features).numpy()
print(output)
# [[-0.2  4.  -1.6]
#  [ 2.  -1.   9. ]]

# MindSpore
import mindspore
from mindspore import Tensor
import mindspore.nn as nn

x = Tensor([[-1.0, 4.0, -8.0], [2.0, -5.0, 9.0]]).astype('float32')
m = nn.LeakyReLU()
output = m(x)
print(output)
# [[-0.2  4.  -1.6]
#  [ 2.  -1.   9. ]]