BasisConv2D

hypercoil.nn.tsconv.BasisConv2D(basis_functions: Sequence[Callable], out_channels: int = 1, memory: int = 3, kernel_width: int = 1, padding: Literal['initial', 'final'] | Sequence[Tuple[int, int]] | None = None, use_bias: bool = False, include_const: bool = False, *, key: jax.random.PRNGKey)

2D convolution over a basis expansion of an input signal.