compute_full_convolution_index#

compute_full_convolution_index(outer_shape, inner_shape, outer_split, inner_split)[source]#

Computes the position of the convolution of pieces in the full convolution.

Parameters:
outer_shapetuple

Tuple of integers corresponding to the shape of the outer array.

inner_shapetuple

Tuple of integers corresponding to the shape of the inner array.

outer_splittuple

Tuple of slices used to split outer array (see split_shape()).

inner_splittuple

Tuple of slices used to split inner array (see split_shape()).

Returns:
tuple

Tuple of slices corresponding to the position of the given convolution in the full convolution.