Convolve

class torchaudio.transforms.Convolve(mode: str = 'full')

Convolves inputs along their last dimension using the direct method. Note that, in contrast to torch.nn.Conv1d, which actually applies the valid cross-correlation operator, this module applies the true convolution operator.

Parameters

mode (str, optional) –

Must be one of (“full”, “valid”, “same”).

• ”full”: Returns the full convolution result, with shape (…, N + M - 1), where N and M are the trailing dimensions of the two inputs. (Default)

• ”valid”: Returns the segment of the full convolution result corresponding to where the two inputs overlap completely, with shape (…, max(N, M) - min(N, M) + 1).

• ”same”: Returns the center segment of the full convolution result, with shape (…, N).

forward(x: Tensor, y: Tensor) → Tensor

Parameters

• x (torch.Tensor) – First convolution operand, with shape (…, N).

• y (torch.Tensor) – Second convolution operand, with shape (…, M) (leading dimensions must be broadcast-able with those of x).

Returns

Result of convolving x and y, with shape (…, L), where the leading dimensions match those of x and L is dictated by mode.

Return type

torch.Tensor