jax.nn.identity#

jax.nn.identity(x)[源代码]#

恒等激活函数。

原样返回参数。

参数:

x (ArrayLike) – 输入数组

返回:

原样返回的参数 x

返回类型:

Array

示例

>>> jax.nn.identity(jax.numpy.array([-2., -1., -0.5, 0, 0.5, 1., 2.]))
Array([-2. , -1. , -0.5, 0. , 0.5, 1. , 2. ], dtype=float32)