cunumeric.convolve#
- cunumeric.convolve(a, v, mode='full')#
Returns the discrete, linear convolution of two ndarrays.
If a and v are both 1-D and v is longer than a, the two are swapped before computation. For N-D cases, the arguments are never swapped.
- Parameters
a ((N,) array_like) – First input ndarray.
v ((M,) array_like) – Second input ndarray.
mode (
{'full', 'valid', 'same'}
, optional) –- ‘same’:
The output is the same size as a, centered with respect to the ‘full’ output. (default)
- ’full’:
The output is the full discrete linear convolution of the inputs.
- ’valid’:
The output consists only of those elements that do not rely on the zero-padding. In ‘valid’ mode, either a or v must be at least as large as the other in every dimension.
- Returns
out – Discrete, linear convolution of a and v.
- Return type
See also
Notes
The current implementation only supports the ‘same’ mode.
Unlike numpy.convolve, cunumeric.convolve supports N-dimensional inputs, but it follows NumPy’s behavior for 1-D inputs.
- Availability
Multiple GPUs, Multiple CPUs