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)