neuraltrain.models.simpleconv.SimpleConvModel¶

class neuraltrain.models.simpleconv.SimpleConvModel(in_channels: int, out_channels: int, config: SimpleConv | None = None)[source][source]¶

nn.Module implementation of SimpleConv.

forward(x: Tensor, subject_ids: Tensor | None = None, channel_positions: Tensor | None = None) → Tensor[source][source]¶

Run the convolutional encoder.

Parameters:

x (Tensor) – Input of shape (B, C, T).
subject_ids (Tensor or None) – Per-example subject indices, shape (B,).
channel_positions (Tensor or None) – Normalised electrode coordinates, shape (B, C, D).