使用pytorch实现IIR低通滤波器的代码

以下是使用 PyTorch 实现 IIR 低通滤波器的代码：

import torch
import torch.nn as nn

class IIRFilter(nn.Module):
    def __init__(self, b, a):
        super(IIRFilter, self).__init__()
        self.b = nn.Parameter(torch.tensor(b, dtype=torch.float32))
        self.a = nn.Parameter(torch.tensor(a, dtype=torch.float32))
        self.zi = nn.Parameter(torch.zeros(len(b) - 1, dtype=torch.float32))

    def forward(self, x):
        y, self.zi = torch.nn.functional.iirfilter(self.b, self.a, x, zi=self.zi)
        return y

希望能对你有所帮助！