帮我写一个经过1*1卷积之后经过空洞卷积的代码

以下是经过1*1卷积之后经过空洞卷积的代码：

import torch.nn as nn

class MyModel(nn.Module):
    def __init__(self):
        super(MyModel, self).__init__()
        self.conv1 = nn.Conv2d(in_channels=3, out_channels=16, kernel_size=1)
        self.conv2 = nn.Conv2d(in_channels=16, out_channels=32, kernel_size=3, dilation=2)

    def forward(self, x):
        x = self.conv1(x)
        x = self.conv2(x)
        return x

希望对您有所帮助。