(self, xs)
return jnp.concatenate(xs, self._axis)
Executes this layer as part of a forward pass through the model.
Executes this layer as part of a forward pass through the model.
Executes this layer as part of a forward pass through the model .
def forward(self, xs):
    """Executes this layer as part of a forward pass through the model."""
    return jnp.concatenate(xs, self._axis)
def forward ( self , xs ) : return jnp . concatenate ( xs , self . _axis )