XLinearNet
chronax.xlinear_module.XLinearNet · inherits nnx.Module
Forward per NF XLinear at n_series=1, no exog. I/O [B, L, 1] -> [B, h, 1].
__init__(self, h: int, input_size: int, hidden_size: int, temporal_ff: int, channel_ff: int, use_norm: bool, *, rngs: nnx.Rngs)
(undocumented)
| Parameter | Type | Default | Description |
|---|---|---|---|
h |
int |
- | (undocumented) |
input_size |
int |
- | (undocumented) |
hidden_size |
int |
- | (undocumented) |
temporal_ff |
int |
- | (undocumented) |
channel_ff |
int |
- | (undocumented) |
use_norm |
bool |
- | (undocumented) |
rngs |
nnx.Rngs |
- | (undocumented) |
__call__(self, x: jnp.ndarray) -> jnp.ndarray
(undocumented)
Parameters:
| Parameter | Type | Default | Description |
|---|---|---|---|
x |
jnp.ndarray |
- | (undocumented) |
Returns: jnp.ndarray (Output shape [B, h, 1]).