jaxsnn.discrete.decode

Classes

DenseData

alias of jax.Array

Functions

jaxsnn.discrete.decode.max_over_time_decode(inputs: jax.Array) → jax.Array

Decode the output of a jaxsnn model by taking the maximum over time and applying a log softmax.

Params inputs

DenseData of shape (n_batch_size, n_time_steps, n_neurons) representing the output of a jaxsnn model.

Returns

jnp.ndarray of shape (n_batch_size, n_neurons) representing the decoded output.