torch_ecg.utils.compute_sequential_output_shape

torch_ecg.utils.compute_sequential_output_shape(model: Sequential, seq_len: int | None = None, batch_size: int | None = None) → Sequence[int | None]

Compute the output shape of a sequential model.

Parameters:

• seq_len (int, optional) – Length of the input tensors.

• batch_size (int, optional) – Batch size of the input tensors.

Returns:

output_shape – The output shape of the module.

Return type:

sequence