fishy.models.utils¶

Shared tensor utilities for model forward passes.

Functions

fishy.models.utils.ensure_conv_input(x: Tensor) → Tensor[source]¶: Ensure a 2D (B, F) tensor becomes (B, 1, F) for Conv1D layers.

fishy.models.utils.ensure_seq_input(x: Tensor) → Tensor[source]¶: Ensure a 2D (B, F) tensor becomes (B, F, 1) for sequence models (LSTM, RWKV, etc.).

s