nanodiffusion代码逐行理解之ResidualBlock

使用了残差连接的两个卷积模块，在第一个卷积之后，增加了时间条件和类条件。
输入:
x: (N, in_channels, H, W)
time_emb: (N, time_emb_dim) or None
y: classes tensor of shape (N) or None
输出:
(N, out_channels, H, W)

class ResidualBlock(nn.Module):
    def __init__(
        self,
        in_channels,
        out_