cupyx.scipy.linalg.convolution_matrix#

cupyx.scipy.linalg.convolution_matrix(a, n, mode='full')[源代码]#

构造一个卷积矩阵。

构造表示一维卷积的 Toeplitz 矩阵。

参数:
  • a (cupy.ndarray) – 要进行卷积的 1-D 数组。

  • n (int) – 结果矩阵的列数。它表示将与 a 进行卷积的输入的长度。这类似于 numpy.convolve(a, v)v 的长度。

  • mode (str) – 这必须是 ('full', 'valid', 'same') 之一。这类似于 numpy.convolve(v, a, mode) 中的 mode

返回:

卷积矩阵,其行数 k 取决于 mode

mode

k

'full'

m + n - 1

'same'

max(m, n)

'valid'

max(m, n) - min(m, n) + 1

返回类型:

cupy.ndarray