帮我用pytorch 写一个拉普拉斯算子

Sure, 这是一个使用PyTorch实现拉普拉斯算子的代码：

import torch

def laplacian_filter(img):
    laplacian_kernel = torch.tensor([[0, 1, 0],
                                     [1, -4, 1],
                                     [0, 1, 0]]).float().unsqueeze(0).unsqueeze(0)

    laplacian = torch.nn.functional.conv2d(img.unsqueeze(0), laplacian_kernel, padding=1, stride=1)
    return laplacian.squeeze(0).squeeze(0)

这段代码实现了一个拉普拉斯滤波器，该滤波器可以通过将其应用于输入图像而获得其拉普拉斯表示。在这个例子中，我们使用了一个3x3的卷积核，其中-4是放在中心位置来强调周围的值相对于中心点重要性的概念。函数返回的是一个相同大小的张量，其中每个像素的值都是拉普拉斯算子的结果。