Identity Channel Mixer¤
discretax.channel_mixers.identity.IdentityChannelMixer
¤
Identity channel mixer.
This channel mixer simply returns the input unchanged.
__init__(in_features: int, key: PRNGKeyArray, *args, **kwargs)
¤
Initialize the identity channel mixer.
Parameters:
| Name | Type | Description | Default |
|---|---|---|---|
in_features
|
int
|
the input dimensionality. |
required |
key
|
PRNGKeyArray
|
JAX random key for initialization. |
required |
*args
|
Additional arguments for the channel mixer. |
required | |
**kwargs
|
Additional keyword arguments for the channel mixer. |
required |
__call__(x: Array) -> Array
¤
Forward pass of the identity channel mixer.
Parameters:
| Name | Type | Description | Default |
|---|---|---|---|
x
|
Array
|
Input tensor. |
required |
Returns:
| Type | Description |
|---|---|
Array
|
The input tensor unchanged. |