fftconv1d_bhl_w_reshape#

fftconv1d_bhl_w_reshape(x, kernel, shortcut=None)#

1D non-causal convolution via subq_ops for BLH inputs [B, L, H].

Reshapes to BHL, runs fftconv1d_bhl(), reshapes back.

Parameters:
  • x (Tensor) – Input tensor [B, L, H].

  • kernel (Tensor) – Kernel tensor [1, K, H]; see fftconv1d_bhl().

  • shortcut (Tensor | None) – Optional per-channel scale [H].

Returns:

Output tensor [B, L, H] in x.dtype.

Return type:

Tensor