Skip to content

Convolve

Parameters concerning the FFT convolution.

batch_size#

batch_size: PositiveInt | None = None

The batch size to use for the FFT convolution.

Since the performing FFT can lead to memory bottlenecks, the convolution can be performed in batches. The batch_size parameter determines the number of matrix elements to compute in each batch. If set to None, the entire convolution is performed in a single batch.