Identity

class brainstate.nn.Identity(name=None)

A placeholder identity operator that is argument-insensitive.

Examples

>>> import brainstate.nn as nn
>>> m = nn.Identity()
>>> x = brainstate.random.randn(2, 3)
>>> output = m(x)
>>> assert (output == x).all()