MLXBackend.compute_convolution_shapes#
- MLXBackend.compute_convolution_shapes(arr1_shape, arr2_shape)#
Computes regular, optimized and fourier convolution shape.
- Parameters:
- arr1_shapetuple
Tuple of integers corresponding to array1 shape.
- arr2_shapetuple
Tuple of integers corresponding to array2 shape.
- Returns:
- tuple
Tuple with regular convolution shape, convolution shape optimized for faster fourier transform, shape of the forward fourier transform (see
build_fft()).